From f585521c05f0f2fa4280d8d8f66fec16c03937e5 Mon Sep 17 00:00:00 2001 From: "Adam J. Stewart" Date: Sun, 25 Feb 2024 18:33:20 +0100 Subject: [PATCH] RandomGeoSampler: optional length is optional --- torchgeo/samplers/single.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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: