using monkeypatch to test predict_step

Author: keves1

tests/trainers/test_change.py

@@ -53,6 +53,9 @@ def test_trainer(
         config = os.path.join('tests', 'conf', name + '.yaml')
 
         monkeypatch.setattr(smp, 'Unet', create_model)
+        monkeypatch.setattr(
+            OSCDDataModule, 'predict_dataloader', OSCDDataModule.test_dataloader
+        )
 
         args = [
             '--config',

torchgeo/datamodules/oscd.py

@@ -108,7 +108,6 @@ def setup(self, stage: str) -> None:
         Args:
             stage: Either 'fit', 'validate', 'test', or 'predict'.
         """
-        self.dataset = OSCD(split='test', **self.kwargs)
         if stage in ['fit', 'validate']:
             self.dataset = OSCD(split='train', **self.kwargs)
             generator = torch.Generator().manual_seed(0)