diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index 6443e9a..500d885 100644 --- 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 index 207205d..3a80c0d 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,4 +1,6 @@ -networkx < 3.0 -numpy -protobuf -torch \ No newline at end of file +torch==2.1.0 +torchvision==0.16.0 +torchaudio==2.1.0 +networkx==2.8.8 +numpy==1.24.1 +protobuf==3.19.4