ruff format

tests/datasets/test_substation.py

@@ -87,9 +87,9 @@ def test_getitem_semantic(self, config: dict[str, Any]) -> None:
 
         x = dataset[0]
         assert isinstance(x, dict), f'Expected dict, got {type(x)}'
-        assert isinstance(
-            x['image'], torch.Tensor
-        ), 'Expected image to be a torch.Tensor'
+        assert isinstance(x['image'], torch.Tensor), (
+            'Expected image to be a torch.Tensor'
+        )
         assert isinstance(x['mask'], torch.Tensor), 'Expected mask to be a torch.Tensor'
 
     def test_len(self, dataset: Substation) -> None:

torchgeo/datamodules/substation.py

@@ -75,15 +75,19 @@ def __init__(
         self.image_resize = image_resize
         self.mask_resize = mask_resize
         self.num_of_timepoints = num_of_timepoints
-        self.geo_transforms = geo_transforms if geo_transforms is not None else self._identity
-        self.color_transforms = color_transforms if color_transforms is not None else self._identity
+        self.geo_transforms = (
+            geo_transforms if geo_transforms is not None else self._identity
+        )
+        self.color_transforms = (
+            color_transforms if color_transforms is not None else self._identity
+        )
         self.image_resize = image_resize if image_resize is not None else self._identity
         self.mask_resize = mask_resize if mask_resize is not None else self._identity
 
     def _identity(self, x: torch.Tensor) -> torch.Tensor:
         """Identity function for default transformations."""
         return x
-    
+
     def setup(self, stage: str) -> None:
         """Set up datasets.