From d93937fc365b9a48794230d8b3463bb7dbf08ad8 Mon Sep 17 00:00:00 2001 From: Ramon Emiliani Date: Wed, 12 May 2021 14:53:46 -0500 Subject: [PATCH] Pass config and task_function as keyword arguments for Hydra's `run_job` change in parameter order. --- clearml/binding/hydra_bind.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/clearml/binding/hydra_bind.py b/clearml/binding/hydra_bind.py index fcb3523d..586240a6 100644 --- a/clearml/binding/hydra_bind.py +++ b/clearml/binding/hydra_bind.py @@ -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