[BUG] Fix actor pool initialization in ray client mode

Eventual-Inc · Oct 11, 2024 · 2234066 · 2234066
1 parent 9ae8122
commit 2234066
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/daft/runners/ray_runner.py b/daft/runners/ray_runner.py
@@ -1173,7 +1173,9 @@ def actor_pool_context(
         if self.ray_client_mode:
             try:
                 yield ray.get(
-                    self.scheduler_actor.get_actor_pool.remote(name, resource_request, num_actors, projection)
+                    self.scheduler_actor.get_actor_pool.remote(
+                        name, resource_request, num_actors, projection, execution_config
+                    )
                 )
             finally:
                 self.scheduler_actor.teardown_actor_pool.remote(name)