Coverage fix

microsoft · Jun 29, 2023 · d83be5e · d83be5e
1 parent ee0df8a
commit d83be5e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/torchgeo/datamodules/vhr10.py b/torchgeo/datamodules/vhr10.py
@@ -17,8 +17,6 @@
 from ..transforms import AugmentationSequential
 from .geo import NonGeoDataModule
 
-# kornia DataKeys must accept MASKS
-
 
 class _AugPipe(Module):
     """Pipeline for applying augmentations sequentially on select data keys."""

diff --git a/torchgeo/datasets/vhr10.py b/torchgeo/datasets/vhr10.py
@@ -39,10 +39,7 @@ def convert_coco_poly_to_mask(
         mask = torch.as_tensor(mask, dtype=torch.uint8)
         mask = mask.any(dim=2)
         masks.append(mask)
-    if masks:
-        masks_tensor = torch.stack(masks, dim=0)
-    else:
-        masks_tensor = torch.zeros((0, height, width), dtype=torch.uint8)
+    masks_tensor = torch.stack(masks, dim=0)
     return masks_tensor