[hotfix] update requirements-test (hpcaitech#701)

edenbuaa · Apr 8, 2022 · 70e8dd4 · 70e8dd4
1 parent 1ae94ea
commit 70e8dd4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -22,13 +22,13 @@ jobs:
         run: |
           pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
           pip install -U pip setuptools wheel --user
-          pip install pytest tensorboard transformers
       - uses: actions/checkout@v2
         with:
           ssh-key: ${{ secrets.SSH_KEY_FOR_CI }}
       - name: Install Colossal-AI      
         run: |
           pip install -r requirements/requirements.txt
+          pip install -r requirements/requirements-test.txt
           pip install -v --no-cache-dir .
       - name: Unit Testing
         run: |

diff --git a/.github/workflows/compatibility_test.yml b/.github/workflows/compatibility_test.yml
@@ -48,11 +48,11 @@ jobs:
         run: |
           pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
           pip install -U pip setuptools wheel --user
-          pip install pytest tensorboard transformers
       - uses: actions/checkout@v2
       - name: Install Colossal-AI      
         run: |
           pip install -r requirements/requirements.txt
+          pip install -r requirements/requirements-test.txt
           pip install -v --no-cache-dir .
       - name: Unit Testing
         run: |

diff --git a/requirements/requirements-test.txt b/requirements/requirements-test.txt
@@ -1,3 +1,5 @@
 pytest
 rpyc
 matplotlib
+tensorboard
+transformers