diff --git a/python/ray/_private/function_manager.py b/python/ray/_private/function_manager.py index 63a0c497e7d84..b492686cf89d8 100644 --- a/python/ray/_private/function_manager.py +++ b/python/ray/_private/function_manager.py @@ -568,9 +568,7 @@ def load_actor_class(self, job_id, actor_creation_function_descriptor): ) method_id = method_descriptor.function_id executor = self._make_actor_method_executor( - actor_method_name, - actor_method, - actor_imported=True, + actor_method_name, actor_method ) self._function_execution_info[method_id] = FunctionExecutionInfo( function=executor, @@ -665,9 +663,7 @@ def _load_actor_class_from_gcs(self, job_id, actor_creation_function_descriptor) actor_class.__module__ = module_name return actor_class - def _make_actor_method_executor( - self, method_name: str, method, actor_imported: bool - ): + def _make_actor_method_executor(self, method_name: str, method): """Make an executor that wraps a user-defined actor method. The wrapped method updates the worker's internal state and performs any necessary checkpointing operations. @@ -676,9 +672,6 @@ def _make_actor_method_executor( method: The actor method to wrap. This should be a method defined on the actor class and should therefore take an instance of the actor as the first argument. - actor_imported: Whether the actor has been imported. - Checkpointing operations will not be run if this is set to - False. Returns: A function that executes the given actor method on the worker's stored instance of the actor. The function also updates the