Fix bug

Signed-off-by: Balaji Veeramani <[email protected]>

python/ray/data/_internal/execution/operators/actor_pool_map_operator.py

@@ -128,7 +128,7 @@ def start(self, options: ExecutionOptions):
         # upstream operators, leading to a spike in memory usage prior to steady state.
         logger.debug(f"{self._name}: Waiting for {len(refs)} pool actors to start...")
         try:
-            timeout = DataContext.get_current().wait_for_min_actors_timeout_s
+            timeout = DataContext.get_current().wait_for_min_actors_s
             ray.get(refs, timeout=timeout)
         except ray.exceptions.GetTimeoutError:
             raise ray.exceptions.GetTimeoutError(

python/ray/data/context.py

@@ -275,7 +275,7 @@ class DataContext:
     )
     print_on_execution_start: bool = True
     s3_try_create_dir: bool = DEFAULT_S3_TRY_CREATE_DIR
-    wait_for_min_actors_s: int = DEFAULT_WAIT_FOR_MIN_ACTORS_S
+    wait_for_min_actors_timeout_s: int = DEFAULT_WAIT_FOR_MIN_ACTORS_S
 
     def __post_init__(self):
         # The additonal ray remote args that should be added to

python/ray/data/tests/test_actor_pool_map_operator.py

@@ -525,7 +525,7 @@ def __call__(self, x):
 
     from ray.exceptions import GetTimeoutError
 
-    ray.data.DataContext.get_current().wait_for_min_actors_timeout_s = 1
+    ray.data.DataContext.get_current().wait_for_min_actors_s = 1
 
     with pytest.raises(
         GetTimeoutError,