mypy fix

microsoft · Jan 10, 2024 · 14acc85 · 14acc85
1 parent ac5f80e
commit 14acc85
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/torchgeo/datasets/vhr10.py b/torchgeo/datasets/vhr10.py
@@ -289,7 +289,8 @@ def _load_image(self, id_: int) -> Tensor:
         )
         with Image.open(filename) as img:
             array: "np.typing.NDArray[np.int_]" = np.array(img)
-            tensor = torch.from_numpy(array).float()
+            tensor = torch.from_numpy(array)
+            tensor = tensor.float()
             # Convert from HxWxC to CxHxW
             tensor = tensor.permute((2, 0, 1))
             return tensor