update requirements

Bluefog-Lib · Apr 7, 2024 · d67a6d7 · d67a6d7
1 parent ca13fe1
commit d67a6d7
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 6 deletions.
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -70,7 +70,9 @@ jobs:
       run: |
         python -m pip install --upgrade pip
         python -m pip install -r requirements.txt
-        python -m pip install -e .
+        python -m pip install networkx 
+        python -c "import networkx"
+        python -m pip install .
         cat tools/requirements_dev.txt | grep pytest | xargs pip install
     - name: Unit Test
       run: |
@@ -94,8 +96,11 @@ jobs:
       run: brew install coreutils  # for timeout command
     - name: Install BlueFogLite
       run: |
+        python3 -m pip install --upgrade pip
         python3 -m pip install -r requirements.txt
-        python3 -m pip install -e .
+        python3 -m pip install networkx 
+        python3 -c "import networkx"
+        python3 -m pip install .
         cat tools/requirements_dev.txt | grep pytest | xargs pip install
     - name: Unit Test
       run: |

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,6 @@
-networkx < 3.0
-numpy
-protobuf
-torch
+torch==2.1.0
+torchvision==0.16.0
+torchaudio==2.1.0
+networkx==2.8.8
+numpy==1.24.1
+protobuf==3.19.4