diff --git a/ray-operator/config/samples/ray-cluster.gke-bucket.yaml b/ray-operator/config/samples/ray-cluster.gke-bucket.yaml new file mode 100644 index 0000000000..cbee9096fa --- /dev/null +++ b/ray-operator/config/samples/ray-cluster.gke-bucket.yaml @@ -0,0 +1,31 @@ +apiVersion: ray.io/v1alpha1 +kind: RayCluster +metadata: + name: raycluster-mini +spec: + rayVersion: '2.6.3' + headGroupSpec: + rayStartParams: + dashboard-host: '0.0.0.0' + template: + spec: + serviceAccountName: my-ksa + nodeSelector: + iam.gke.io/gke-metadata-server-enabled: "true" + containers: + - name: ray-head + image: rayproject/ray:2.6.3 + resources: + limits: + cpu: 1 + memory: 2Gi + requests: + cpu: 1 + memory: 2Gi + ports: + - containerPort: 6379 + name: gcs-server + - containerPort: 8265 + name: dashboard + - containerPort: 10001 + name: client diff --git a/ray-operator/config/samples/ray-job.batch-inference.yaml b/ray-operator/config/samples/ray-job.batch-inference.yaml index 3b7b476b6b..a97dc0e97f 100644 --- a/ray-operator/config/samples/ray-job.batch-inference.yaml +++ b/ray-operator/config/samples/ray-job.batch-inference.yaml @@ -72,7 +72,7 @@ data: class ImageClassifier: def __init__(self): # If doing CPU inference, set `device="cpu"` instead. - self.classifier = pipeline("image-classification", model="google/vit-base-patch16-224", device=0) # TODO:archit + self.classifier = pipeline("image-classification", model="google/vit-base-patch16-224", device=0) def __call__(self, batch: Dict[str, np.ndarray]): # Convert the numpy array of images into a list of PIL images which is the format the HF pipeline expects. diff --git a/tests/test_sample_raycluster_yamls.py b/tests/test_sample_raycluster_yamls.py index 7476c89b2e..313ccc68ab 100644 --- a/tests/test_sample_raycluster_yamls.py +++ b/tests/test_sample_raycluster_yamls.py @@ -51,6 +51,7 @@ 'ray-cluster.autoscaler.large.yaml': 'Skip this test because it requires a lot of resources.', 'ray-cluster-tpu.yaml': 'Skip this test because it requires TPU resources.', + 'ray-cluster.gke-bucket.yaml': 'Skip this test because it requires GKE and k8s service accounts.', } rs = RuleSet([HeadPodNameRule(), EasyJobRule(), HeadSvcRule()])