From 738be75a38b8289eda176b634bf9ac1ce187b329 Mon Sep 17 00:00:00 2001 From: Ravin Kohli Date: Wed, 9 Mar 2022 18:31:38 +0100 Subject: [PATCH] fixes after rebase --- autoPyTorch/pipeline/components/training/trainer/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/autoPyTorch/pipeline/components/training/trainer/__init__.py b/autoPyTorch/pipeline/components/training/trainer/__init__.py index c63524fc5..b380659da 100755 --- 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 )