Skip to content

Commit

Permalink
[Runtimes] Fix Pickling in MLRun Function Due to Improper Dynamic Imp…
Browse files Browse the repository at this point in the history
…ort (#5636)
  • Loading branch information
quaark committed May 27, 2024
1 parent 8be29c2 commit 5b36daa
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions mlrun/runtimes/local.py
Original file line number Diff line number Diff line change
Expand Up @@ -384,6 +384,7 @@ def load_module(file_name, handler, context):
if spec is None:
raise RunError(f"Cannot import from {file_name!r}")
module = imputil.module_from_spec(spec)
sys.modules[mod_name] = module
spec.loader.exec_module(module)

class_args = {}
Expand Down

0 comments on commit 5b36daa

Please sign in to comment.