From 169527eaefe9f5485b0efbba2b717607ab42a799 Mon Sep 17 00:00:00 2001 From: Tim Kucera Date: Wed, 24 Apr 2024 12:01:26 +0200 Subject: [PATCH] adjust build reqs --- .github/workflows/build.yml | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index c8b906e7..8f73dcb5 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -25,7 +25,8 @@ jobs: run: | python -m pip install --upgrade pip pip install flake8 pytest Cython - pip install torch torch_geometric tensorflow dgl + pip install torch --index-url https://download.pytorch.org/whl/cpu + pip install torch_geometric tensorflow-cpu dgl if [ -f requirements.txt ]; then pip install -r requirements.txt; fi - name: Test with pytest run: |