ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-04-08 22:54:44 +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)
if not PatchLIGHTgbmModelIO.__main_task:
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():
if isinstance(v, set):
params[k] = list(v)
PatchLIGHTgbmModelIO.__main_task.connect(params)
if params:
PatchLIGHTgbmModelIO.__main_task.connect(params)
return ret