update setup.py and ci install

maps-as-data · Oct 24, 2024 · d24d0bc · d24d0bc
1 parent b6cd879
commit d24d0bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/.github/workflows/mr_ci_text_spotting.yml b/.github/workflows/mr_ci_text_spotting.yml
@@ -33,7 +33,6 @@ jobs:
       - name: Install dependencies
         run: |
           python -m pip install wheel ninja
-          python -m pip install torch==2.2.2 torchvision==0.17.2 -f https://download.pytorch.org/whl/torch_stable.html
           python -m pip install ".[dev, text]" # all dependencies
           python -m pip install pytest-cov
 

diff --git a/setup.py b/setup.py
@@ -38,8 +38,8 @@
         "nest-asyncio>=1.5.1,<2.0.0",
         "scikit-image>=0.18.3",
         "scikit-learn>=1.0.1,<2.0.0",
-        "torch>=1.10.0",
-        "torchvision>=0.11.1,<0.17.3",
+        "torch>=1.10.0,=<2.2.2",
+        "torchvision>=0.11.1,=<0.17.2",
         "jupyter>=1.0.0,<2.0.0",
         "ipykernel>=6.5.1,<7.0.0",
         "ipywidgets>=8.0.0,<9.0.0",