You are working on the torchgeo/torchgeo repository. The source code is checked out at /testbed at commit…

You are working on the torchgeo/torchgeo repository. The source code is checked out at /testbed at commit…: a task in SWE-Lego-RL-2699 (Harbor dataset). code: python sampler = RandomBatchGeoSampler(ds, size=1024, batch size=5, length=5 5) dl = DataLoader(ds, batch sampler=sampler, collate fn=stack…

The task

code: ```python sampler = RandomBatchGeoSampler(ds, size=1024, batch_size=5, length=5 * 5) dl = DataLoader(ds, batch_sampler=sampler, collate_fn=stack_samples)

Part of Lego-X/Lego-RL-2699.