Fix/pytorch vulnerability update (#32)

iiconocimiento · Jul 30, 2024 · e6e42c5 · e6e42c5
1 parent 3b42233
commit e6e42c5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/.github/workflows/unit-tests.yml b/.github/workflows/unit-tests.yml
@@ -42,7 +42,7 @@ jobs:
           pip install poetry
           poetry install --with dev
       - name: Run ruff
-        run: poetry run ruff .
+        run: poetry run ruff check .
 
   codecov:
     needs: unit_test

diff --git a/pyproject.toml b/pyproject.toml
@@ -29,7 +29,7 @@ pandas = ">= 1"
 Pillow = ">= 8.4.0"
 scikit-learn = ">= 1"
 timm = ">= 0.5.4"
-torch = ">= 1.13.1, <=2.0.0"
+torch = ">2.2.0"
 torchvision = ">= 0.11.3"
 
 [tool.poetry.group.dev.dependencies]
@@ -104,7 +104,7 @@ select = [
     "RUF",
 ]
 unfixable = ["B007"]
-ignore = ["E501", "D100", "D103", "D104", "PLR0913", "PLR0915", "PLR2004", "RUF013", "S101", "ANN101"]
+ignore = ["E501", "D100", "D103", "D104", "PLR0912", "PLR0913", "PLR0915", "PLR2004", "RUF013", "S101", "ANN101"]
 
 src = ["aisee", "tests"]