Skip to main content

vantage_sdk.workbench.ray_job

SDK helpers for cluster-scoped Ray job/cluster operations (KubeRay).

Classes

RayJobSDK()

SDK for cluster-scoped Ray job + cluster operations.

Methods

  • create(self, ctx, *, cluster_name: 'str', name: 'str', entrypoint: 'str', namespace: 'str | None' = None, configuration_preset: 'str | None' = None, size_preset: 'str | None' = None, overrides: 'dict[str, Any] | None' = None) -> 'ClusterServiceResponse': Submit a RayJob.
  • delete(self, ctx, *, cluster_name: 'str', name: 'str') -> 'ClusterServiceResponse': Delete a RayJob (KubeRay tears down the associated RayCluster).
  • get(self, ctx, *, cluster_name: 'str', name: 'str') -> 'ClusterServiceResponse': Get one RayJob by name.
  • get_cluster(self, ctx, *, cluster_name: 'str', name: 'str') -> 'ClusterServiceResponse': Get one RayCluster by name.
  • list(self, ctx, *, cluster_name: 'str') -> 'ClusterServiceResponse': List the caller's RayJobs.
  • list_clusters(self, ctx, *, cluster_name: 'str') -> 'ClusterServiceResponse': List the caller's RayClusters (job-owned and standalone).