test

Author: kryanbeane

.github/workflows/e2e_tests.yaml .github/workflows/e2e_tests.yml.github/workflows/e2e_tests.yaml renamed to .github/workflows/e2e_tests.yml

@@ -3,14 +3,9 @@ name: e2e
 
 on:
   pull_request:
-    branches:
-      - main
-      - 'release-*'
-    paths-ignore:
-      - 'docs/**'
-      - '**.adoc'
-      - '**.md'
-      - 'LICENSE'
+    branches: [ main ]
+  push:
+    branches: [ main ]
 
 concurrency:
   group: ${{ github.head_ref }}-${{ github.workflow }}
@@ -117,7 +112,7 @@ jobs:
           pip install poetry
           poetry install --with test,docs
           echo "Running e2e tests..."
-          poetry run pytest -v -s ./tests/e2e -m 'kind and nvidia_gpu' > ${CODEFLARE_TEST_OUTPUT_DIR}/pytest_output.log 2>&1
+          poetry run pytest -v -s ./tests/e2e/local_interactive_sdk_kind_test.py::TestRayLocalInteractiveOauth::test_local_interactives_nvidia_gpu > ${CODEFLARE_TEST_OUTPUT_DIR}/pytest_output.log 2>&1
         env:
           GRPC_DNS_RESOLVER: "native"
 

tests/e2e/local_interactive_sdk_kind_test.py

@@ -48,6 +48,7 @@ def run_local_interactives(
                 head_cpu_limits="500m",
                 head_memory_requests=2,
                 head_memory_limits=2,
+                head_extended_resource_requests={gpu_resource_name: 0},
                 worker_cpu_requests="500m",
                 worker_cpu_limits=1,
                 worker_memory_requests=1,
@@ -68,22 +69,15 @@ def run_local_interactives(
         ray.shutdown()
         ray.init(address=cluster.local_client_url(), logging_level="DEBUG")
 
-        @ray.remote(num_gpus=number_of_gpus / 2)
-        def heavy_calculation_part(num_iterations):
+        @ray.remote(num_gpus=number_of_gpus)
+        def heavy_calculation(num_iterations):
             result = 0.0
             for i in range(num_iterations):
                 for j in range(num_iterations):
                     for k in range(num_iterations):
                         result += math.sin(i) * math.cos(j) * math.tan(k)
             return result
 
-        @ray.remote(num_gpus=number_of_gpus / 2)
-        def heavy_calculation(num_iterations):
-            results = ray.get(
-                [heavy_calculation_part.remote(num_iterations // 30) for _ in range(30)]
-            )
-            return sum(results)
-
         ref = heavy_calculation.remote(3000)
         result = ray.get(ref)
         assert result == 1789.4644387076714