Fix style tests

microsoft · Jan 28, 2025 · 4080d03 · 4080d03
1 parent aff5aaa
commit 4080d03
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/tests/datasets/test_ssl4eo.py b/tests/datasets/test_ssl4eo.py
@@ -63,9 +63,8 @@ def test_getitem(self, dataset: SSL4EOL) -> None:
         x = dataset[0]
         assert isinstance(x, dict)
         assert isinstance(x['image'], torch.Tensor)
-        assert (
-            x['image'].size(0)
-            == dataset.seasons * len(dataset.metadata[dataset.split]['all_bands'])
+        assert x['image'].size(0) == dataset.seasons * len(
+            dataset.metadata[dataset.split]['all_bands']
         )
 
     def test_len(self, dataset: SSL4EOL) -> None:

diff --git a/torchgeo/datasets/ssl4eo.py b/torchgeo/datasets/ssl4eo.py
@@ -232,7 +232,7 @@ def __init__(
         self._verify()
 
         if split.startswith('tm'):
-            base = Landsat5TM
+            base: type[Landsat] = Landsat5TM
         elif split.startswith('etm'):
             base = Landsat7
         else: