Skip to content

Commit

Permalink
Remove print statements
Browse files Browse the repository at this point in the history
  • Loading branch information
PGijsbers committed Sep 17, 2024
1 parent 5e792ac commit 0cb6ca7
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions frameworks/AutoGluon/exec.py
Original file line number Diff line number Diff line change
Expand Up @@ -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


Expand Down

0 comments on commit 0cb6ca7

Please sign in to comment.