From 8a834af777d7c4d1541573158d627c9d39f5c7c5 Mon Sep 17 00:00:00 2001 From: bedapisl Date: Sun, 15 Oct 2023 11:28:54 +0200 Subject: [PATCH] Support lightgbm model loading from a string instead of file (#1136) Co-authored-by: Bedrich Pisl --- clearml/binding/frameworks/lightgbm_bind.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/clearml/binding/frameworks/lightgbm_bind.py b/clearml/binding/frameworks/lightgbm_bind.py index 3493027a..0b842153 100644 --- a/clearml/binding/frameworks/lightgbm_bind.py +++ b/clearml/binding/frameworks/lightgbm_bind.py @@ -71,7 +71,7 @@ class PatchLIGHTgbmModelIO(PatchBaseModelIO): return ret @staticmethod - def _load(original_fn, model_file, *args, **kwargs): + def _load(original_fn, model_file=None, *args, **kwargs): if not PatchLIGHTgbmModelIO._current_task: return original_fn(model_file, *args, **kwargs)