From dd38fddcbdd8c552fa9b9e69a63c7f7586c358b2 Mon Sep 17 00:00:00 2001
From: "Adam J. Stewart" <ajstewart426@gmail.com>
Date: Sun, 3 Mar 2024 00:34:31 +0100
Subject: [PATCH] mypy fixes

---
 tests/models/test_rcf.py     | 2 +-
 torchgeo/datamodules/seco.py | 2 +-
 torchgeo/datasets/vhr10.py   | 4 ++--
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/tests/models/test_rcf.py b/tests/models/test_rcf.py
index e5f6862a14b..bc871e1843d 100644
--- 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
index 613789a0dba..b859e4c8971 100644
--- 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
index 28b95197796..83add564be1 100644
--- 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(