From fd83f8c2cbb47d8f77bf3d3859b1b69d96d2f6ca Mon Sep 17 00:00:00 2001 From: J Alexander Date: Wed, 22 Sep 2021 02:34:05 -0500 Subject: [PATCH] Fix xgb train overload (#456) Co-authored-by: Johnathan Alexander --- clearml/binding/frameworks/xgboost_bind.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clearml/binding/frameworks/xgboost_bind.py b/clearml/binding/frameworks/xgboost_bind.py index 4cc418b1..fb408d00 100644 --- a/clearml/binding/frameworks/xgboost_bind.py +++ b/clearml/binding/frameworks/xgboost_bind.py @@ -116,13 +116,13 @@ class PatchXGBoostModelIO(PatchBaseModelIO): return model @staticmethod - def _train(original_fn, __obj, *args, **kwargs): + def _train(original_fn, *args, **kwargs): if PatchXGBoostModelIO.__callback_cls: callbacks = kwargs.get('callbacks') or [] kwargs['callbacks'] = callbacks + [ PatchXGBoostModelIO.__callback_cls(task=PatchXGBoostModelIO.__main_task) ] - return original_fn(__obj, *args, **kwargs) + return original_fn(*args, **kwargs) @classmethod def _generate_training_callback_class(cls):