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

Pass config and task_function as keyword arguments for Hydra's run_job change in parameter order.

Browse Source
This commit is contained in:
Ramon Emiliani 2021-05-12 14:53:46 -05:00 committed by Allegro AI
parent 55b214c23d
commit d93937fc36
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/binding/hydra_bind.py
View File

@ -118,9 +118,9 @@ class PatchHydra(object):
'Task.init called outside of Hydra-App. For full Hydra multi-run support, '
'move the Task.init call into the Hydra-App main function')
result = PatchHydra._original_run_job(
config, partial(PatchHydra._patched_task_function, task_function,),
*args, **kwargs)
kwargs["config"] = config
kwargs["task_function"] = partial(PatchHydra._patched_task_function, task_function,)
result = PatchHydra._original_run_job(*args, **kwargs)
# if we have Task.init called inside the App, we close it after the app is done.
# This will make sure that hydra run will create multiple Tasks