fixes after rebase

automl · Mar 9, 2022 · 738be75 · 738be75
1 parent 9d622db
commit 738be75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autoPyTorch/pipeline/components/training/trainer/__init__.py b/autoPyTorch/pipeline/components/training/trainer/__init__.py
@@ -351,7 +351,7 @@ def _fit(self, X: Dict[str, Any], y: Any = None, **kwargs: Any) -> 'TrainerChoic
             scheduler=X['lr_scheduler'],
             task_type=STRING_TO_TASK_TYPES[X['dataset_properties']['task_type']],
             labels=labels,
-            step_interval=X['step_interval']
+            step_interval=X['step_interval'],
             numerical_columns=X['dataset_properties']['numerical_columns'] if 'numerical_columns' in X[
                 'dataset_properties'] else None
         )