return argument train dataloader

torchgeo/datamodules/deepglobelandcover.py

@@ -177,7 +177,7 @@ def pad_to(sample: Dict[str, Any]) -> Dict[str, Any]:
             split="test", transforms=test_transforms, **self.kwargs
         )
 
-    def train_dataloader(self) -> DataLoader[Any]:
+    def train_dataloader(self) -> DataLoader[Dict[str, Any]]:
         """Return a DataLoader for training.
 
         Returns: