check version #2

mllam · Jun 3, 2024 · e689650 · e689650
1 parent 7797cef
commit e689650
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/ci-pdm-install-and-test.yml b/.github/workflows/ci-pdm-install-and-test.yml
@@ -30,7 +30,7 @@ jobs:
       - name: Print and check torch version
         run: |
           pdm run python -c "import torch; print(torch.__version__)"
-          pdm run python -c "import torch; assert torch.__version__.endswith('+gpu')"
+          pdm run python -c "import torch; assert torch.__version__.endswith('+cpu')"
 
       - name: Install package (including dev dependencies)
         run: |