From 4e8fae579085c3ad2e4d7225a349242aa6e4f2c0 Mon Sep 17 00:00:00 2001 From: Teddy Koker Date: Fri, 29 Dec 2023 15:22:41 -0500 Subject: [PATCH] use cpu version for testing --- .github/workflows/test.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index b70b13a..3f62442 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -18,6 +18,6 @@ jobs: with: python-version: ${{ matrix.python-version }} - name: Install Dependencies - run: pip install torch && pip install -e '.[testing]' + run: pip install torch --index-url https://download.pytorch.org/whl/cpu && pip install -e '.[testing]' - name: Run Pytest run: python -m pytest -s -v