pass addition kwargs to ray init

B612-Asteroid-Institute · Dec 16, 2023 · e6173ed · e6173ed
1 parent ed08b8d
commit e6173ed
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/adam_core/ray_cluster.py b/adam_core/ray_cluster.py
@@ -7,7 +7,7 @@
 
 
 def initialize_use_ray(
-    num_cpus: Optional[int] = None, object_store_bytes: Optional[int] = None
+    num_cpus: Optional[int] = None, object_store_bytes: Optional[int] = None, **kwargs
 ) -> bool:
     """
     Ensures we use existing local cluster, or starts new one with desired resources
@@ -34,6 +34,7 @@ def initialize_use_ray(
                     dashboard_host="0.0.0.0",
                     num_cpus=num_cpus,
                     object_store_memory=object_store_bytes,
+                    **kwargs,
                 )
 
         logger.info(f"Ray Resources: {ray.cluster_resources()}")