Fixed styling issues

microsoft · Feb 12, 2024 · 2a10b78 · 2a10b78
1 parent 79dc5db
commit 2a10b78
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 24 deletions.
diff --git a/torchgeo/datamodules/l7irish.py b/torchgeo/datamodules/l7irish.py
@@ -68,11 +68,9 @@ def setup(self, stage: str) -> None:
         """
         dataset = L7Irish(**self.kwargs)
         generator = torch.Generator().manual_seed(0)
-        (
-            self.train_dataset,
-            self.val_dataset,
-            self.test_dataset,
-        ) = random_bbox_assignment(dataset, [0.6, 0.2, 0.2], generator)
+        (self.train_dataset, self.val_dataset, self.test_dataset) = (
+            random_bbox_assignment(dataset, [0.6, 0.2, 0.2], generator)
+        )
 
         if stage in ["fit"]:
             self.train_batch_sampler = RandomBatchGeoSampler(

diff --git a/torchgeo/datamodules/l8biome.py b/torchgeo/datamodules/l8biome.py
@@ -68,11 +68,9 @@ def setup(self, stage: str) -> None:
         """
         dataset = L8Biome(**self.kwargs)
         generator = torch.Generator().manual_seed(0)
-        (
-            self.train_dataset,
-            self.val_dataset,
-            self.test_dataset,
-        ) = random_bbox_assignment(dataset, [0.6, 0.2, 0.2], generator)
+        (self.train_dataset, self.val_dataset, self.test_dataset) = (
+            random_bbox_assignment(dataset, [0.6, 0.2, 0.2], generator)
+        )
 
         if stage in ["fit"]:
             self.train_batch_sampler = RandomBatchGeoSampler(

diff --git a/torchgeo/datamodules/nccm.py b/torchgeo/datamodules/nccm.py
@@ -68,11 +68,9 @@ def setup(self, stage: str) -> None:
         generator = torch.Generator().manual_seed(0)
 
         self.dataset = self.sentinel2 & self.nccm
-        (
-            self.train_dataset,
-            self.val_dataset,
-            self.test_dataset,
-        ) = random_bbox_assignment(self.dataset, [0.8, 0.1, 0.1], generator)
+        (self.train_dataset, self.val_dataset, self.test_dataset) = (
+            random_bbox_assignment(self.dataset, [0.8, 0.1, 0.1], generator)
+        )
 
         if stage in ["fit"]:
             self.train_batch_sampler = RandomBatchGeoSampler(

diff --git a/torchgeo/datasets/idtrees.py b/torchgeo/datasets/idtrees.py
@@ -407,14 +407,12 @@ def _load_geometries(self, directory: str) -> dict[int, dict[str, Any]]:
     @overload
     def _filter_boxes(
         self, image_size: tuple[int, int], min_size: int, boxes: Tensor, labels: Tensor
-    ) -> tuple[Tensor, Tensor]:
-        ...
+    ) -> tuple[Tensor, Tensor]: ...
 
     @overload
     def _filter_boxes(
         self, image_size: tuple[int, int], min_size: int, boxes: Tensor, labels: None
-    ) -> tuple[Tensor, None]:
-        ...
+    ) -> tuple[Tensor, None]: ...
 
     def _filter_boxes(
         self,

diff --git a/torchgeo/samplers/utils.py b/torchgeo/samplers/utils.py
@@ -12,13 +12,11 @@
 
 
 @overload
-def _to_tuple(value: Union[tuple[int, int], int]) -> tuple[int, int]:
-    ...
+def _to_tuple(value: Union[tuple[int, int], int]) -> tuple[int, int]: ...
 
 
 @overload
-def _to_tuple(value: Union[tuple[float, float], float]) -> tuple[float, float]:
-    ...
+def _to_tuple(value: Union[tuple[float, float], float]) -> tuple[float, float]: ...
 
 
 def _to_tuple(value: Union[tuple[float, float], float]) -> tuple[float, float]:

diff --git a/torchgeo/trainers/segmentation.py b/torchgeo/trainers/segmentation.py
@@ -9,7 +9,6 @@
 
 import matplotlib.pyplot as plt
 import segmentation_models_pytorch as smp
-import torch
 import torch.nn as nn
 from matplotlib.figure import Figure
 from torch import Tensor