Env var for custom Ray Image

Author: ChristianZaccaria

tests/e2e/mnist_raycluster_sdk_test.py

@@ -46,7 +46,7 @@ def create_test_namespace(self):
         return self.namespace
 
     def run_mnist_raycluster_sdk(self):
-        ray_image = "quay.io/project-codeflare/ray:latest-py39-cu118"
+        ray_image = os.getenv("RAY_IMAGE")
         host = os.getenv("CLUSTER_HOSTNAME")
 
         ingress_options = {}

tests/e2e/support.py

@@ -1,14 +1,12 @@
+import os
 import random
 import string
 
-def read_file(file_name):
-    try:
-        with open(file_name, 'rb') as file:
-            return file.read()
-    except IOError as e:
-        raise e
+def get_ray_image():
+    default_ray_image = "quay.io/project-codeflare/ray:latest-py39-cu118"
+    return os.getenv("RAY_IMAGE", default_ray_image)
 
 
-alphabet = string.ascii_lowercase + string.digits
 def random_choice():
+    alphabet = string.ascii_lowercase + string.digits
     return ''.join(random.choices(alphabet, k=5))