ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix LightGBM support for empty params (precaution)

Browse Source
This commit is contained in:
allegroai 2020-11-25 11:22:35 +02:00
parent 28d7527537
commit 333c46e9c9
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
trains/binding/frameworks/lightgbm_bind.py
View File

@ -123,9 +123,10 @@ class PatchLIGHTgbmModelIO(PatchBaseModelIO):
ret = original_fn(*args, **kwargs) ret = original_fn(*args, **kwargs)
if not PatchLIGHTgbmModelIO.__main_task: if not PatchLIGHTgbmModelIO.__main_task:
return ret return ret
params = args[0] if args else kwargs['params'] params = args[0] if args else kwargs.get('params', {})
for k, v in params.items(): for k, v in params.items():
if isinstance(v, set): if isinstance(v, set):
params[k] = list(v) params[k] = list(v)
PatchLIGHTgbmModelIO.__main_task.connect(params) if params:
PatchLIGHTgbmModelIO.__main_task.connect(params)
return ret return ret