diff --git a/torchgeo/samplers/single.py b/torchgeo/samplers/single.py index 7251f4b274f..360798b583d 100644 --- 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: