ray remote resources as plain dict (#1737)

facebookresearch · Aug 24, 2021 · b16baa7 · b16baa7
1 parent e3ba7cc
commit b16baa7
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/plugins/hydra_ray_launcher/hydra_plugins/hydra_ray_launcher/_launcher_util.py b/plugins/hydra_ray_launcher/hydra_plugins/hydra_ray_launcher/_launcher_util.py
@@ -10,7 +10,7 @@
 from hydra.core.singleton import Singleton
 from hydra.core.utils import JobReturn, run_job, setup_globals
 from hydra.types import HydraContext, TaskFunction
-from omegaconf import DictConfig
+from omegaconf import DictConfig, OmegaConf
 
 # mypy complains about "unused type: ignore comment" on macos
 # workaround adapted from: https://github.com/twisted/twisted/pull/1416
@@ -64,7 +64,9 @@ def launch_job_on_ray(
     singleton_state: Any,
 ) -> Any:
     if ray_remote:
-        run_job_ray = ray.remote(**ray_remote)(_run_job)
+        run_job_ray = ray.remote(**OmegaConf.to_container(ray_remote, resolve=True))(
+            _run_job
+        )
     else:
         run_job_ray = ray.remote(_run_job)