From 148d47733c7b695dcb12ceecc127576a7f9cc60c Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Tue, 26 Nov 2024 14:34:07 +0000 Subject: [PATCH 1/2] Pin mypy --- .github/workflows/lint.yml | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/lint.yml b/.github/workflows/lint.yml index 7e9943668cd..22ada97fba8 100644 --- a/.github/workflows/lint.yml +++ b/.github/workflows/lint.yml @@ -72,7 +72,7 @@ jobs: script: | set -euo pipefail - export PYTHON_VERSION=3.9 + export PYTHON_VERSION=3.11 export GPU_ARCH_TYPE=cpu export GPU_ARCH_VERSION='' @@ -83,7 +83,7 @@ jobs: conda activate ci echo '::group::Install lint tools' - pip install --progress-bar=off mypy + pip install --progress-bar=off "mypy==1.13.0" echo '::endgroup::' echo '::group::Lint Python types' From 6ddd1c2f2b791631e4c1d9829f1f24ba53e6595c Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Tue, 26 Nov 2024 14:48:21 +0000 Subject: [PATCH 2/2] Fix --- torchvision/models/maxvit.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchvision/models/maxvit.py b/torchvision/models/maxvit.py index 2dbdca69f05..2a3888b2af3 100644 --- a/torchvision/models/maxvit.py +++ b/torchvision/models/maxvit.py @@ -688,7 +688,7 @@ def __init__( p_stochastic=p_stochastic[p_idx : p_idx + num_layers], ), ) - input_size = self.blocks[-1].grid_size + input_size = self.blocks[-1].grid_size # type: ignore[assignment] p_idx += num_layers # see https://github.com/google-research/maxvit/blob/da76cf0d8a6ec668cc31b399c4126186da7da944/maxvit/models/maxvit.py#L1137-L1158