Skip to content

Commit 1da4ea3

Browse files
bveeramanielliot-barn
authored andcommitted
[Data] Fix type annotation for actor_location_tracker arg of _MapWorker (#57859)
## Description The type annotation for `actor_location_tracker` is currently `ActorLocationTracker`, but it should be `ray.actor.ActorHandle[ActorLocationTracker]`. This PR fixes that issue. Signed-off-by: Balaji Veeramani <[email protected]> Signed-off-by: elliot-barn <[email protected]>
1 parent dfe742c commit 1da4ea3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -543,7 +543,7 @@ def __init__(
543543
src_fn_name: str,
544544
map_transformer: MapTransformer,
545545
logical_actor_id: str,
546-
actor_location_tracker: ActorLocationTracker,
546+
actor_location_tracker: ray.actor.ActorHandle[ActorLocationTracker],
547547
):
548548
self.src_fn_name: str = src_fn_name
549549
self._map_transformer = map_transformer

0 commit comments

Comments
 (0)