Reverting

microsoft · Mar 22, 2023 · 7e78884 · 7e78884
1 parent ce19188
commit 7e78884
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/tests/datasets/test_geo.py b/tests/datasets/test_geo.py
@@ -182,9 +182,7 @@ def sentinel(self, request: SubRequest) -> Sentinel2:
     @pytest.fixture()
     def custom_dtype_ds(self) -> RasterDataset:
         root = os.path.join("tests", "data", "raster")
-        ds = RasterDataset(root)
-        ds.dtype = torch.long
-        return ds
+        return RasterDataset(root)
 
     def test_getitem_single_file(self, naip: NAIP) -> None:
         x = naip[naip.bounds]

diff --git a/torchgeo/datasets/geo.py b/torchgeo/datasets/geo.py
@@ -435,7 +435,7 @@ def __getitem__(self, query: BoundingBox) -> Dict[str, Any]:
             data = data.to(self.dtype)
 
         if self.is_image:
-            sample["image"] = data
+            sample["image"] = data.float()
         else:
             sample["mask"] = data