Remove post-execute hook for ray task

flyteorg · Mar 28, 2024 · f1a11b8 · f1a11b8
1 parent 8ab9a3c
commit f1a11b8
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/plugins/flytekit-ray/flytekitplugins/ray/task.py b/plugins/flytekit-ray/flytekitplugins/ray/task.py
@@ -56,10 +56,6 @@ def pre_execute(self, user_params: ExecutionParameters) -> ExecutionParameters:
         ray.init(address=self._task_config.address)
         return user_params
 
-    def post_execute(self, user_params: ExecutionParameters, rval: Any) -> Any:
-        ray.shutdown()
-        return rval
-
     def get_custom(self, settings: SerializationSettings) -> Optional[Dict[str, Any]]:
         cfg = self._task_config