pytorch-tabular
diff --git a/‎src/pytorch_tabular/tabular_model.py‎
Lines changed: 1 addition & 1 deletion b/‎src/pytorch_tabular/tabular_model.py‎
Lines changed: 1 addition & 1 deletion
@@ -1498,7 +1498,7 @@ def add_noise(module, input, output):
                     ret_logits,
                     include_input_features=False,
                     device=device,
-                    progress_bar=progress_bar or "None",
+                    progress_bar=progress_bar or "none",
                 )
                 pred_idx = pred_df.index
                 if self.config.task == "classification":
Original file line number	Diff line number	Diff line change
`@@ -1498,7 +1498,7 @@ def add_noise(module, input, output):`
`1498`	`1498`	`ret_logits,`
`1499`	`1499`	`include_input_features=False,`
`1500`	`1500`	`device=device,`
`1501`		`- progress_bar=progress_bar or "None",`
	`1501`	`+ progress_bar=progress_bar or "none",`
`1502`	`1502`	`)`
`1503`	`1503`	`pred_idx = pred_df.index`
`1504`	`1504`	`if self.config.task == "classification":`