Run only 1 test

andreyvelich · Sep 14, 2023 · ad1c34e · ad1c34e
1 parent 27094ee
commit ad1c34e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/integration-tests.yaml b/.github/workflows/integration-tests.yaml
@@ -109,7 +109,7 @@ jobs:
       - name: Run tests
         run: |
           pip install pytest
-          python3 -m pip install -e sdk/python; pytest sdk/python/test --log-cli-level=info --namespace=default
+          python3 -m pip install -e sdk/python; pytest sdk/python/test --log-cli-level=info --namespace=default -k "test_pytorchjob_from_func"
         env:
           GANG_SCHEDULER_NAME: ${{ matrix.gang-scheduler-name }}
 

diff --git a/sdk/python/test/e2e/test_e2e_pytorchjob.py b/sdk/python/test/e2e/test_e2e_pytorchjob.py
@@ -166,7 +166,7 @@ def train_func(parameters):
         namespace=job_namespace,
         parameters={"lr": "0.01"},
         train_func=train_func,
-        base_image="gcr.io/kubeflow-ci/pytorch-dist-mnist-test:v1.0",
+        base_image="python:3.8-slim",
         num_worker_replicas=1,
         packages_to_install=["pandas==1.3.5"],
     )