NumPy 2 support (#2151)

pyproject.toml

@@ -268,7 +268,7 @@ quote-style = "single"
 skip-magic-trailing-comma = true
 
 [tool.ruff.lint]
-extend-select = ["D", "I", "UP"]
+extend-select = ["D", "I", "NPY201", "UP"]
 
 [tool.ruff.lint.per-file-ignores]
 "docs/**" = ["D"]

torchgeo/datasets/biomassters.py

@@ -196,7 +196,7 @@ def _load_target(self, filename: str) -> Tensor:
             target mask
         """
         with rasterio.open(os.path.join(self.root, 'train_agbm', filename), 'r') as src:
-            arr: np.typing.NDArray[np.float_] = src.read()
+            arr: np.typing.NDArray[np.float64] = src.read()
 
         target = torch.from_numpy(arr).float()
         return target

torchgeo/datasets/dfc2022.py

@@ -235,7 +235,7 @@ def _load_image(self, path: str, shape: Sequence[int] | None = None) -> Tensor:
             the image
         """
         with rasterio.open(path) as f:
-            array: np.typing.NDArray[np.float_] = f.read(
+            array: np.typing.NDArray[np.float64] = f.read(
                 out_shape=shape, out_dtype='float32', resampling=Resampling.bilinear
             )
             tensor = torch.from_numpy(array)

torchgeo/datasets/eurocrops.py

@@ -243,10 +243,12 @@ def plot(
 
         fig, axs = plt.subplots(nrows=1, ncols=ncols, figsize=(4, 4))
 
-        def apply_cmap(arr: 'np.typing.NDArray[Any]') -> 'np.typing.NDArray[np.float_]':
+        def apply_cmap(
+            arr: 'np.typing.NDArray[Any]',
+        ) -> 'np.typing.NDArray[np.float64]':
             # Color 0 as black, while applying default color map for the class indices.
             cmap = plt.get_cmap('viridis')
-            im: np.typing.NDArray[np.float_] = cmap(arr / len(self.class_map))
+            im: np.typing.NDArray[np.float64] = cmap(arr / len(self.class_map))
             im[arr == 0] = 0
             return im