flyteplugins-ray 2.0.0b7__py3-none-any.whl → 2.0.0b30__py3-none-any.whl

flyteplugins/ray/task.py

@@ -5,12 +5,12 @@ import typing
 from dataclasses import dataclass
 from typing import Any, Dict, Optional
 
+import flyte
 import yaml
 from flyte import PodTemplate, Resources
-from flyte._tools import is_in_cluster
 from flyte.extend import AsyncFunctionTaskTemplate, TaskPluginRegistry, pod_spec_from_resources
 from flyte.models import SerializationContext
-from flyteidl.plugins.ray_pb2 import HeadGroupSpec, RayCluster, RayJob, WorkerGroupSpec
+from flyteidl2.plugins.ray_pb2 import HeadGroupSpec, RayCluster, RayJob, WorkerGroupSpec
 from google.protobuf.json_format import MessageToDict
 
 import ray
@@ -62,15 +62,16 @@ class RayFunctionTask(AsyncFunctionTaskTemplate):
 
     task_type: str = "ray"
     plugin_config: RayJobConfig
+    debuggable: bool = True
 
     async def pre(self, *args, **kwargs) -> Dict[str, Any]:
         init_params = {"address": self.plugin_config.address}
 
-        if is_in_cluster():
+        if flyte.ctx().is_in_cluster():
             working_dir = os.getcwd()
             init_params["runtime_env"] = {
                 "working_dir": working_dir,
-                "excludes": ["script_mode.tar.gz", "fast*.tar.gz", ".python_history"],
+                "excludes": ["script_mode.tar.gz", "fast*.tar.gz", ".python_history", ".code-server"],
             }
 
         if not ray.is_initialized():

{flyteplugins_ray-2.0.0b7.dist-info → flyteplugins_ray-2.0.0b30.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flyteplugins-ray
-Version: 2.0.0b7
+Version: 2.0.0b30
 Summary: Ray plugin for flyte
 Author-email: Kevin Su <pingsutw@users.noreply.github.com>
 Requires-Python: >=3.10
@@ -19,5 +19,5 @@ a type of cluster spun up for a specific Ray job and torn down after completion.
 To install the plugin, run the following command:
 
 ```bash
-pip install flyteplguins-ray
+pip install --pre flyteplugins-ray
 ```

flyteplugins_ray-2.0.0b30.dist-info/RECORD

@@ -0,0 +1,6 @@
+flyteplugins/ray/__init__.py,sha256=6FlHBUczAg13S-Lkd1DhwmjMzznpM_wYLemyOEkH1IE,147
+flyteplugins/ray/task.py,sha256=SijShR7VrApXwlH8CgfD9pBlgS72xZ5noYXyCO7QDTs,5151
+flyteplugins_ray-2.0.0b30.dist-info/METADATA,sha256=tfyFt8tpCku01f0z0xzZ1ObFJJnT60FfqtszZ1XqvHI,740
+flyteplugins_ray-2.0.0b30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flyteplugins_ray-2.0.0b30.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
+flyteplugins_ray-2.0.0b30.dist-info/RECORD,,

flyteplugins_ray-2.0.0b7.dist-info/RECORD

@@ -1,6 +0,0 @@
-flyteplugins/ray/__init__.py,sha256=6FlHBUczAg13S-Lkd1DhwmjMzznpM_wYLemyOEkH1IE,147
-flyteplugins/ray/task.py,sha256=iOF5rKtW8KhZrGDkMLLIlpDK_Wv4VYFSAEiygoXsxdg,5120
-flyteplugins_ray-2.0.0b7.dist-info/METADATA,sha256=E0qMsRhRYQ0dDxLv8SDZvNN_vpkTZdMWSEefEWtCmE8,733
-flyteplugins_ray-2.0.0b7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flyteplugins_ray-2.0.0b7.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
-flyteplugins_ray-2.0.0b7.dist-info/RECORD,,