Bump the torch group in /requirements with 2 updates (#2546)

* Bump the torch group in /requirements with 2 updates Bumps the torch group in /requirements with 2 updates: [torch](https://github.com/pytorch/pytorch) and [torchvision](https://github.com/pytorch/vision). Updates `torch` from 2.5.1 to 2.6.0 - [Release notes](https://github.com/pytorch/pytorch/releases) - [Changelog](https://github.com/pytorch/pytorch/blob/main/RELEASE.md) - [Commits](pytorch/pytorch@v2.5.1...v2.6.0) Updates `torchvision` from 0.20.1 to 0.21.0 - [Release notes](https://github.com/pytorch/vision/releases) - [Commits](pytorch/vision@v0.20.1...v0.21.0) --- updated-dependencies: - dependency-name: torch dependency-type: direct:production update-type: version-update:semver-minor dependency-group: torch - dependency-name: torchvision dependency-type: direct:production update-type: version-update:semver-minor dependency-group: torch ... Signed-off-by: dependabot[bot] <[email protected]> * Fix mypy --------- Signed-off-by: dependabot[bot] <[email protected]> Co-authored-by: dependabot[bot] <49699333+dependabot[bot]@users.noreply.github.com> Co-authored-by: Adam J. Stewart <[email protected]>
microsoft · Jan 30, 2025 · b259ebf · b259ebf
1 parent ddb2b64
commit b259ebf
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -29,7 +29,7 @@ repos:
           - pillow>=10.4.0
           - pytest>=6.1.2
           - scikit-image>=0.22.0
-          - torch>=2.3
+          - torch>=2.6
           - torchmetrics>=0.10
           - torchvision>=0.18
         exclude: (build|data|dist|logo|logs|output)/

diff --git a/requirements/required.txt b/requirements/required.txt
@@ -17,6 +17,6 @@ rtree==1.3.0
 segmentation-models-pytorch==0.4.0
 shapely==2.0.6
 timm==1.0.14
-torch==2.5.1
+torch==2.6.0
 torchmetrics==1.6.1
-torchvision==0.20.1
+torchvision==0.21.0
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -27,4 +27,4 @@ def matplotlib_backend() -> None:
 
 @pytest.fixture(autouse=True)
 def torch_hub(tmp_path: Path) -> None:
-    torch.hub.set_dir(tmp_path)  # type: ignore[no-untyped-call]
+    torch.hub.set_dir(tmp_path)
diff --git a/tests/models/test_croma.py b/tests/models/test_croma.py
@@ -18,7 +18,7 @@
 )
 
 
-def save_model(model: torch.nn.Module, path: Path) -> None:
+def save_model(model: CROMA, path: Path) -> None:
     state_dict = {
         's1_encoder': model.s1_encoder.state_dict(),
         's1_GAP_FFN': model.s1_GAP_FFN.state_dict(),