diff --git a/frameworks/AutoGluon/exec.py b/frameworks/AutoGluon/exec.py index 7cff5fd2d..6e16277e5 100644 --- a/frameworks/AutoGluon/exec.py +++ b/frameworks/AutoGluon/exec.py @@ -176,19 +176,16 @@ def inference_time_regression(data: Union[str, pd.DataFrame]): def initialize_callbacks(callback_settings): - print(callback_settings) callbacks = [] try: import autogluon.core.callbacks except ImportError: raise ValueError("Callbacks are only available for AutoGluon>=1.1.2") - print(callbacks) for callback, hyperparameters in callback_settings.items(): callback_cls = getattr(autogluon.core.callbacks, callback, None) if not callback_cls: raise ValueError(f"Callback {callback} is not a valid callback") callbacks.append(callback_cls(**hyperparameters)) - print(callbacks) return callbacks