RandomGeoSampler: optional length is optional (#1907)

microsoft · Mar 2, 2024 · 76c46a4 · 76c46a4
1 parent 1395525
commit 76c46a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/samplers/single.py b/torchgeo/samplers/single.py
@@ -70,7 +70,7 @@ def __init__(
         self,
         dataset: GeoDataset,
         size: Union[tuple[float, float], float],
-        length: Optional[int],
+        length: Optional[int] = None,
         roi: Optional[BoundingBox] = None,
         units: Units = Units.PIXELS,
     ) -> None: