Resource request and limits

project-codeflare · Oct 5, 2023 · 24c16a8 · 24c16a8
1 parent b4aee3a
commit 24c16a8
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 19 deletions.
diff --git a/src/codeflare_sdk/templates/base-template.yaml b/src/codeflare_sdk/templates/base-template.yaml
@@ -16,21 +16,21 @@ spec:
       custompodresources:
       - replicas: 1
         requests:
-          cpu: 2
-          memory: 8G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 0
         limits:
-          cpu: 2
-          memory: 8G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 0
       - replicas: 3
         requests:
-          cpu: 2
-          memory: 12G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 1
         limits:
-          cpu: 2
-          memory: 12G
+          cpu: 1
+          memory: 4G
           nvidia.com/gpu: 1
       generictemplate:
         # This config demonstrates KubeRay's Ray autoscaler integration.

diff --git a/tests/e2e/codeflare_sdk-0.0.0.dev0-py3-none-any.whl b/tests/e2e/codeflare_sdk-0.0.0.dev0-py3-none-any.whl
diff --git a/tests/e2e/mnist_raycluster_sdk.py b/tests/e2e/mnist_raycluster_sdk.py
@@ -9,17 +9,19 @@
 
 namespace = sys.argv[1]
 
-cluster = Cluster(ClusterConfiguration(
-    name='mnist',
-    namespace=namespace,
-    num_workers=1,
-    min_cpus='500m',
-    max_cpus=1,
-    min_memory=0.5,
-    max_memory=1,
-    num_gpus=0,
-    instascale=False,
-))
+cluster = Cluster(
+    ClusterConfiguration(
+        name="mnist",
+        namespace=namespace,
+        num_workers=1,
+        min_cpus="500m",
+        max_cpus=1,
+        min_memory=0.5,
+        max_memory=1,
+        num_gpus=0,
+        instascale=False,
+    )
+)
 
 cluster.up()