test: change deprecated cluster up to apply and add more resources to ray clusters

Author: pawelpaszki

tests/e2e/heterogeneous_clusters_kind_test.py

@@ -63,7 +63,7 @@ def run_heterogeneous_clusters(
                     local_queue=queue_name,
                 )
             )
-            cluster.up()
+            cluster.apply()
             sleep(5)
             node_name = get_pod_node(self, self.namespace, cluster_name)
             print(f"Cluster {cluster_name}-{flavor} is running on node: {node_name}")

tests/e2e/heterogeneous_clusters_oauth_test.py

@@ -55,9 +55,9 @@ def run_heterogeneous_clusters(
                     namespace=self.namespace,
                     name=cluster_name,
                     num_workers=1,
-                    head_cpu_requests="500m",
-                    head_cpu_limits="500m",
-                    worker_cpu_requests="500m",
+                    head_cpu_requests=1,
+                    head_cpu_limits=1,
+                    worker_cpu_requests=1,
                     worker_cpu_limits=1,
                     worker_memory_requests=1,
                     worker_memory_limits=4,
@@ -66,7 +66,7 @@ def run_heterogeneous_clusters(
                     local_queue=queue_name,
                 )
             )
-            cluster.up()
+            cluster.apply()
             sleep(5)
             node_name = get_pod_node(self, self.namespace, cluster_name)
             print(f"Cluster {cluster_name}-{flavor} is running on node: {node_name}")

tests/e2e/local_interactive_sdk_kind_test.py

@@ -65,7 +65,7 @@ def run_local_interactives(
             )
         )
 
-        cluster.up()
+        cluster.apply()
 
         cluster.wait_ready()
         cluster.status()

tests/e2e/local_interactive_sdk_oauth_test.py

@@ -44,6 +44,10 @@ def run_local_interactives(self):
                 namespace=self.namespace,
                 name=cluster_name,
                 num_workers=1,
+                head_memory_requests=6,
+                head_memory_limits=8,
+                head_cpu_requests=1,
+                head_cpu_limits=1,
                 worker_cpu_requests=1,
                 worker_cpu_limits=1,
                 worker_memory_requests=1,
@@ -52,7 +56,7 @@ def run_local_interactives(self):
                 verify_tls=False,
             )
         )
-        cluster.up()
+        cluster.apply()
         cluster.wait_ready()
 
         generate_cert.generate_tls_cert(cluster_name, self.namespace)

tests/e2e/mnist_raycluster_sdk_aw_kind_test.py

@@ -55,7 +55,7 @@ def run_mnist_raycluster_sdk_kind(
             )
         )
 
-        cluster.up()
+        cluster.apply()
 
         cluster.status()
 

tests/e2e/mnist_raycluster_sdk_kind_test.py

@@ -54,7 +54,7 @@ def run_mnist_raycluster_sdk_kind(
             )
         )
 
-        cluster.up()
+        cluster.apply()
 
         cluster.status()
 

tests/e2e/mnist_raycluster_sdk_oauth_test.py

@@ -46,19 +46,19 @@ def run_mnist_raycluster_sdk_oauth(self):
                 name="mnist",
                 namespace=self.namespace,
                 num_workers=1,
-                head_cpu_requests="500m",
-                head_cpu_limits="500m",
+                head_memory_requests=6,
+                head_memory_limits=8,
                 worker_cpu_requests=1,
                 worker_cpu_limits=1,
-                worker_memory_requests=1,
-                worker_memory_limits=4,
+                worker_memory_requests=6,
+                worker_memory_limits=8,
                 image=ray_image,
                 write_to_file=True,
                 verify_tls=False,
             )
         )
 
-        cluster.up()
+        cluster.apply()
 
         cluster.status()
 

tests/e2e/start_ray_cluster.py

@@ -26,7 +26,7 @@
     )
 )
 
-cluster.up()
+cluster.apply()
 
 cluster.status()