ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-04 02:57:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix xgb train overload (#456)

Browse Source 
Co-authored-by: Johnathan Alexander <jalexander86@gatech.edu>
This commit is contained in:
J Alexander 2021-09-22 02:34:05 -05:00 committed by GitHub
parent 6c96e60174
commit fd83f8c2cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/binding/frameworks/xgboost_bind.py
View File

@ -116,13 +116,13 @@ class PatchXGBoostModelIO(PatchBaseModelIO):
return model return model
@staticmethod @staticmethod
def _train(original_fn, __obj, *args, **kwargs): def _train(original_fn, *args, **kwargs):
if PatchXGBoostModelIO.__callback_cls: if PatchXGBoostModelIO.__callback_cls:
callbacks = kwargs.get('callbacks') or [] callbacks = kwargs.get('callbacks') or []
kwargs['callbacks'] = callbacks + [ kwargs['callbacks'] = callbacks + [
PatchXGBoostModelIO.__callback_cls(task=PatchXGBoostModelIO.__main_task) PatchXGBoostModelIO.__callback_cls(task=PatchXGBoostModelIO.__main_task)
] ]
return original_fn(__obj, *args, **kwargs) return original_fn(*args, **kwargs)
@classmethod @classmethod
def _generate_training_callback_class(cls): def _generate_training_callback_class(cls):