mypy fixes

microsoft · Mar 2, 2024 · dd38fdd · dd38fdd
1 parent 238d586
commit dd38fdd
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/tests/models/test_rcf.py b/tests/models/test_rcf.py
@@ -38,7 +38,7 @@ def test_untrainable(self) -> None:
     def test_biases(self) -> None:
         model = RCF(features=24, bias=10, mode="gaussian")
         # https://github.com/pytorch/pytorch/issues/116328
-        assert torch.all(model.biases == 10)  # type: ignore[call-overload]
+        assert torch.all(model.biases == 10)
 
     def test_seed(self) -> None:
         weights1 = RCF(seed=1, mode="gaussian").weights

diff --git a/torchgeo/datamodules/seco.py b/torchgeo/datamodules/seco.py
@@ -51,7 +51,7 @@ def __init__(
 
             self.aug = AugmentationSequential(
                 # https://github.com/arogozhnikov/einops/issues/299
-                K.Normalize(mean=_min, std=_max - _min),  # type: ignore[operator]
+                K.Normalize(mean=_min, std=_max - _min),
                 K.Normalize(mean=torch.tensor(0), std=1 / torch.tensor(255)),
                 K.Normalize(mean=_mean, std=_std),
                 data_keys=["image"],

diff --git a/torchgeo/datasets/vhr10.py b/torchgeo/datasets/vhr10.py
@@ -472,7 +472,7 @@ def plot(
             # Add masks
             if show_feats in {"masks", "both"} and "masks" in sample:
                 mask = masks[i]
-                contours = find_contours(mask, 0.5)
+                contours = find_contours(mask, 0.5)  # type: ignore[no-untyped-call]
                 for verts in contours:
                     verts = np.fliplr(verts)
                     p = patches.Polygon(
@@ -524,7 +524,7 @@ def plot(
                 # Add masks
                 if show_pred_masks:
                     mask = prediction_masks[i]
-                    contours = find_contours(mask, 0.5)
+                    contours = find_contours(mask, 0.5)  # type: ignore[no-untyped-call]
                     for verts in contours:
                         verts = np.fliplr(verts)
                         p = patches.Polygon(