Added source pytorch_cpu to pyproject.toml

mivanit · Jul 24, 2023 · 7af61d6 · 7af61d6
1 parent 39e8109
commit 7af61d6
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/.github/workflows/checks.yml b/.github/workflows/checks.yml
@@ -52,7 +52,7 @@ jobs:
           curl -sSL https://install.python-poetry.org | python3 -
           poetry lock --check
           export CUDA_VISIBLE_DEVICES=0
-          poetry add torch@${{ matrix.versions.torch }}+cpu --source https://download.pytorch.org/whl/cpu
+          poetry add torch@${{ matrix.versions.torch }}+cpu --source torch_cpu
           poetry install --all-extras
 
       - name: tests

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -38,6 +38,11 @@ matplotlib = "^3.0.0"
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
 
+[[tool.poetry.source]]
+name = "torch_cpu"
+url = "https://download.pytorch.org/whl/cpu"
+priority = "explicit"
+
 # TODO: make all of the following ignored across all formatting/linting
 # tests/input_data, tests/junk_data, muutils/_wip