Update tests

tests/datasets/test_so2sat.py

@@ -21,13 +21,15 @@
 class TestSo2Sat:
     @pytest.fixture(params=["train", "validation", "test"])
     def dataset(self, monkeypatch: MonkeyPatch, request: SubRequest) -> So2Sat:
-        md5s = {
-            "train": "82e0f2d51766b89cb905dbaf8275eb5b",
-            "validation": "bf292ae4737c1698b1a3c6f5e742e0e1",
-            "test": "9a3bbe181b038d4e51f122c4be3c569e",
+        md5s_by_version = {
+            "2": {
+                "train": "82e0f2d51766b89cb905dbaf8275eb5b",
+                "validation": "bf292ae4737c1698b1a3c6f5e742e0e1",
+                "test": "9a3bbe181b038d4e51f122c4be3c569e",
+            }
         }
 
-        monkeypatch.setattr(So2Sat, "md5s", md5s)
+        monkeypatch.setattr(So2Sat, "md5s_by_version", md5s_by_version)
         root = os.path.join("tests", "data", "so2sat")
         split = request.param
         transforms = nn.Identity()

torchgeo/datasets/so2sat.py

@@ -82,44 +82,33 @@ class So2Sat(NonGeoDataset):
 
        or manually downloaded from https://mediatum.ub.tum.de/1613658
     """  # noqa: E501
+
     versions = ["2", "3_random", "3_block", "3_culture_10"]
     filenames_by_version = {
         "2": {
             "train": "training.h5",
             "validation": "validation.h5",
             "test": "testing.h5",
         },
-        "3_random": {
-            "train": "random/training.h5",
-            "test": "random/testing.h5",
-        },
-        "3_block": {
-            "train": "block/training.h5",
-            "test": "block/testing.h5",
-        },
+        "3_random": {"train": "random/training.h5", "test": "random/testing.h5"},
+        "3_block": {"train": "block/training.h5", "test": "block/testing.h5"},
         "3_culture_10": {
             "train": "culture_10/training.h5",
             "test": "culture_10/testing.h5",
-        }
+        },
     }
     md5s_by_version = {
         "2": {
             "train": "702bc6a9368ebff4542d791e53469244",
             "validation": "71cfa6795de3e22207229d06d6f8775d",
             "test": "e81426102b488623a723beab52b31a8a",
         },
-        "3_random": {
-            "train": "",
-            "test": "",
-        },
-        "3_block": {
-            "train": "",
-            "test": "",
-        },
+        "3_random": {"train": "", "test": ""},
+        "3_block": {"train": "", "test": ""},
         "3_culture_10": {
             "train": "702bc6a9368ebff4542d791e53469244",
             "test": "58335ce34ca3a18424e19da84f2832fc",
-        }
+        },
     }
 
     classes = [