Fix typo

microsoft · Dec 26, 2022 · 29cec5d · 29cec5d
1 parent 15f57d2
commit 29cec5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datamodules/loveda.py b/torchgeo/datamodules/loveda.py
@@ -78,7 +78,7 @@ def setup(self, stage: Optional[str] = None) -> None:
         )
 
         self.predict_dataset = LoveDA(
-            split="test", transforms=val_pred_transforms, **self.kwargs
+            split="test", transforms=val_predict_transforms, **self.kwargs
         )
 
     def train_dataloader(self) -> DataLoader[Any]: