Updating test case

microsoft · Mar 22, 2023 · ce19188 · ce19188
1 parent 6a992cf
commit ce19188
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/datasets/test_geo.py b/tests/datasets/test_geo.py
@@ -182,7 +182,9 @@ def sentinel(self, request: SubRequest) -> Sentinel2:
     @pytest.fixture()
     def custom_dtype_ds(self) -> RasterDataset:
         root = os.path.join("tests", "data", "raster")
-        return RasterDataset(root)
+        ds = RasterDataset(root)
+        ds.dtype = torch.long
+        return ds
 
     def test_getitem_single_file(self, naip: NAIP) -> None:
         x = naip[naip.bounds]