diff --git a/experiments/ssl4eo/landsat/chip_landsat_benchmark.py b/experiments/ssl4eo/landsat/chip_landsat_benchmark.py index b8ec0bb61f5..77bfaa0f057 100755 --- a/experiments/ssl4eo/landsat/chip_landsat_benchmark.py +++ b/experiments/ssl4eo/landsat/chip_landsat_benchmark.py @@ -65,9 +65,10 @@ def retrieve_mask_chip( layer_name = "cdl" for img_path in tqdm(paths): - with rasterio.open(img_path) as img_src, rasterio.open( - args.mask_path - ) as mask_src: + with ( + rasterio.open(img_path) as img_src, + rasterio.open(args.mask_path) as mask_src, + ): if mask_src.crs != img_src.crs: mask_src = WarpedVRT(mask_src, crs=img_src.crs) diff --git a/torchgeo/datamodules/l7irish.py b/torchgeo/datamodules/l7irish.py index 573bc996528..dbc6dddebb0 100644 --- 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 index be14f2e1221..f372aba051b 100644 --- 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/datasets/idtrees.py b/torchgeo/datasets/idtrees.py index 2d3833cf1b3..b743e5e81be 100644 --- a/torchgeo/datasets/idtrees.py +++ b/torchgeo/datasets/idtrees.py @@ -406,14 +406,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/datasets/ssl4eo_benchmark.py b/torchgeo/datasets/ssl4eo_benchmark.py index 59314a8fb68..d292d0894ec 100644 --- a/torchgeo/datasets/ssl4eo_benchmark.py +++ b/torchgeo/datasets/ssl4eo_benchmark.py @@ -245,9 +245,11 @@ def _download(self) -> None: download_url( self.url.format(self.mask_dir_name), self.root, - md5=self.mask_md5s[self.sensor.split("_")[0]][self.product] - if self.checksum - else None, + md5=( + self.mask_md5s[self.sensor.split("_")[0]][self.product] + if self.checksum + else None + ), ) def _extract(self) -> None: diff --git a/torchgeo/models/farseg.py b/torchgeo/models/farseg.py index a1faae16a5b..2dfc3c01894 100644 --- a/torchgeo/models/farseg.py +++ b/torchgeo/models/farseg.py @@ -219,9 +219,11 @@ def __init__( ), BatchNorm2d(out_channels), ReLU(inplace=True), - UpsamplingBilinear2d(scale_factor=2) - if num_upsample != 0 - else Identity(), + ( + UpsamplingBilinear2d(scale_factor=2) + if num_upsample != 0 + else Identity() + ), ) for idx in range(num_layers) ] diff --git a/torchgeo/samplers/utils.py b/torchgeo/samplers/utils.py index 9c05cc867b4..329f3677e0b 100644 --- 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]: