Test custom CML runner

FR-DC · Nov 8, 2023 · 56fb64c · 56fb64c
1 parent 1064c34
commit 56fb64c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.github/workflows/model.yml b/.github/workflows/model.yml
@@ -9,8 +9,8 @@ jobs:
   build:
 
     runs-on: self-hosted
-    container:
-      image: nvidia/cuda:12.2.2-devel-ubuntu22.04
+#    container:
+#      image: nvidia/cuda:12.2.2-devel-ubuntu22.04
     steps:
       - uses: actions/checkout@v3
 
@@ -45,6 +45,6 @@ jobs:
 
       - name: Test with pytest through poetry
         run: |
-          poetry run pytest tests/model_tests
+          poetry run pytest