We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 366bede commit 637a68bCopy full SHA for 637a68b
autoPyTorch/pipeline/components/training/trainer/__init__.py
@@ -323,7 +323,7 @@ def prepare_trainer(self, X: Dict) -> None:
323
scheduler=X['lr_scheduler'],
324
task_type=STRING_TO_TASK_TYPES[X['dataset_properties']['task_type']],
325
labels=labels,
326
- step_interval=X['step_interval']
+ step_interval=X['step_interval'],
327
numerical_columns=X['dataset_properties']['numerical_columns'] if 'numerical_columns' in X[
328
'dataset_properties'] else None
329
)
0 commit comments