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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. flyteplugins/ray/task.py +3 -2
  2. {flyteplugins_ray-2.0.0b14.dist-info → flyteplugins_ray-2.0.0b30.dist-info}/METADATA +1 -1
  3. flyteplugins_ray-2.0.0b30.dist-info/RECORD +6 -0
  4. flyteplugins_ray-2.0.0b14.dist-info/RECORD +0 -6
  5. {flyteplugins_ray-2.0.0b14.dist-info → flyteplugins_ray-2.0.0b30.dist-info}/WHEEL +0 -0
  6. {flyteplugins_ray-2.0.0b14.dist-info → flyteplugins_ray-2.0.0b30.dist-info}/top_level.txt +0 -0
flyteplugins/ray/task.py CHANGED
@@ -10,7 +10,7 @@ import yaml
10
10
  from flyte import PodTemplate, Resources
11
11
  from flyte.extend import AsyncFunctionTaskTemplate, TaskPluginRegistry, pod_spec_from_resources
12
12
  from flyte.models import SerializationContext
13
- from flyteidl.plugins.ray_pb2 import HeadGroupSpec, RayCluster, RayJob, WorkerGroupSpec
13
+ from flyteidl2.plugins.ray_pb2 import HeadGroupSpec, RayCluster, RayJob, WorkerGroupSpec
14
14
  from google.protobuf.json_format import MessageToDict
15
15
 
16
16
  import ray
@@ -62,6 +62,7 @@ class RayFunctionTask(AsyncFunctionTaskTemplate):
62
62
 
63
63
  task_type: str = "ray"
64
64
  plugin_config: RayJobConfig
65
+ debuggable: bool = True
65
66
 
66
67
  async def pre(self, *args, **kwargs) -> Dict[str, Any]:
67
68
  init_params = {"address": self.plugin_config.address}
@@ -70,7 +71,7 @@ class RayFunctionTask(AsyncFunctionTaskTemplate):
70
71
  working_dir = os.getcwd()
71
72
  init_params["runtime_env"] = {
72
73
  "working_dir": working_dir,
73
- "excludes": ["script_mode.tar.gz", "fast*.tar.gz", ".python_history"],
74
+ "excludes": ["script_mode.tar.gz", "fast*.tar.gz", ".python_history", ".code-server"],
74
75
  }
75
76
 
76
77
  if not ray.is_initialized():
{flyteplugins_ray-2.0.0b14.dist-info → flyteplugins_ray-2.0.0b30.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: flyteplugins-ray
3
- Version: 2.0.0b14
3
+ Version: 2.0.0b30
4
4
  Summary: Ray plugin for flyte
5
5
  Author-email: Kevin Su <pingsutw@users.noreply.github.com>
6
6
  Requires-Python: >=3.10
flyteplugins_ray-2.0.0b30.dist-info/RECORD ADDED
@@ -0,0 +1,6 @@
1
+ flyteplugins/ray/__init__.py,sha256=6FlHBUczAg13S-Lkd1DhwmjMzznpM_wYLemyOEkH1IE,147
2
+ flyteplugins/ray/task.py,sha256=SijShR7VrApXwlH8CgfD9pBlgS72xZ5noYXyCO7QDTs,5151
3
+ flyteplugins_ray-2.0.0b30.dist-info/METADATA,sha256=tfyFt8tpCku01f0z0xzZ1ObFJJnT60FfqtszZ1XqvHI,740
4
+ flyteplugins_ray-2.0.0b30.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
5
+ flyteplugins_ray-2.0.0b30.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
6
+ flyteplugins_ray-2.0.0b30.dist-info/RECORD,,
flyteplugins_ray-2.0.0b14.dist-info/RECORD DELETED
@@ -1,6 +0,0 @@
1
- flyteplugins/ray/__init__.py,sha256=6FlHBUczAg13S-Lkd1DhwmjMzznpM_wYLemyOEkH1IE,147
2
- flyteplugins/ray/task.py,sha256=we3Pfvczr8-eAZAL7wzYaq5wFIcQ0Dgv68Tc7qoYoZ8,5106
3
- flyteplugins_ray-2.0.0b14.dist-info/METADATA,sha256=qwoExcsEBVFZF0WvGHGiH_0lUfS0ghROWpezKJcwlR8,740
4
- flyteplugins_ray-2.0.0b14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
5
- flyteplugins_ray-2.0.0b14.dist-info/top_level.txt,sha256=cgd779rPu9EsvdtuYgUxNHHgElaQvPn74KhB5XSeMBE,13
6
- flyteplugins_ray-2.0.0b14.dist-info/RECORD,,