Update dependencies (#651)

adapter-hub · Feb 17, 2024 · 69ea0b4 · 69ea0b4
1 parent 29cbdd7
commit 69ea0b4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/.github/workflows/tests_torch.yml b/.github/workflows/tests_torch.yml
@@ -39,7 +39,7 @@ jobs:
           key: ${{ runner.os }}-pip-${{ hashFiles('setup.py') }}
       - name: Install
         run: |
-          pip install torch==2.1.1
+          pip install torch==2.1.2
           pip install .[quality]
       - name: Check Quality and Repo Consistency
         run: |
@@ -62,7 +62,7 @@ jobs:
             ${{ runner.os }}-pip-
       - name: Install
         run: |
-          pip install torch==2.1.1
+          pip install torch==2.1.2
           pip install .[sklearn,testing,sentencepiece]
       - name: Test
         run: |
@@ -85,7 +85,7 @@ jobs:
             ${{ runner.os }}-pip-
       - name: Install
         run: |
-          pip install torch==2.1.1
+          pip install torch==2.1.2
           pip install .[sklearn,testing,sentencepiece]
       - name: Test
         run: |

diff --git a/setup.py b/setup.py
@@ -37,9 +37,9 @@
     "onnxruntime>=1.4.0",
     "parameterized",
     "pillow",
-    "protobuf<=3.20.2",
+    "protobuf",
     "psutil",
-    "pytest",
+    "pytest>=7.2.0,<8.0.0",
     "pytest-subtests",
     "pytest-timeout",
     "pytest-xdist",
@@ -88,6 +88,7 @@ def deps_list(*pkgs):
 extras["sentencepiece"] = deps_list("sentencepiece", "protobuf")
 extras["testing"] = deps_list(
     "pytest",
+    "pytest-subtests",
     "pytest-xdist",
     "timeout-decorator",
     "parameterized",
@@ -101,7 +102,6 @@ def deps_list(*pkgs):
     "rouge-score",
     "nltk",
     "GitPython",
-    "protobuf",  # Can be removed once we can unpin protobuf
     "sacremoses",
     "rjieba",
     "beautifulsoup4",