adjusted pipeline

gebauerm · Oct 20, 2024 · b72fb8c · b72fb8c
1 parent 989db7d
commit b72fb8c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.github/workflows/python-app.yml b/.github/workflows/python-app.yml
@@ -24,6 +24,10 @@ jobs:
       uses: actions/setup-python@v3
       with:
         python-version: ${{ matrix.python-version }}
+    # TODO: pytorch install has to be done in an extra github job: see pytorch_geometric for an example
+    - name: Install torch
+      run: /
+        pip3 install torch==1.11.0 torchvision==0.12.0
     - name: Install package
       run: |
         python -m pip install --upgrade pip

diff --git a/rapidflow/__init__.py b/rapidflow/__init__.py
@@ -7,7 +7,7 @@
 install_requires = [
         "optuna>=2.9.1",
         "click>=8.0.1", "scikit-learn", "scipy", "networkx>=2.5.1", "psycopg2-binary", "docker>=5.0.3",
-        "pandas", "tqdm>=4.62.3", "torch==2.1.0"],
+        "pandas", "tqdm>=4.62.3"],
 
 test_require = ["pytest==7.1.2", "pytest-cov==3.0.0"]