add predict

Author: awaelchli

tests/trainer/test_dataloaders.py

@@ -636,7 +636,7 @@ def test_warning_with_few_workers_multi_loader(_, tmpdir, ckpt_path, stage):
 
 def test_warning_with_iterable_dataset_and_len(tmpdir):
     """ Tests that a warning message is shown when an IterableDataset defines `__len__`. """
-    model = EvalModelTemplate()
+    model = BoringModel()
     original_dataset = model.train_dataloader().dataset
 
     class IterableWithoutLen(IterableDataset):
@@ -660,6 +660,8 @@ def __len__(self):
         trainer.fit(model, train_dataloader=dataloader, val_dataloaders=[dataloader])
     with pytest.warns(UserWarning, match='Your `IterableDataset` has `__len__` defined.'):
         trainer.test(model, test_dataloaders=[dataloader])
+    with pytest.warns(UserWarning, match='Your `IterableDataset` has `__len__` defined.'):
+        trainer.predict(model, dataloaders=[dataloader])
 
     # without __len__ defined
     dataloader = DataLoader(IterableWithoutLen(), batch_size=16)
@@ -669,6 +671,7 @@ def __len__(self):
     trainer.validate(model, val_dataloaders=dataloader)
     trainer.fit(model, train_dataloader=dataloader, val_dataloaders=[dataloader])
     trainer.test(model, test_dataloaders=dataloader)
+    trainer.predict(model, dataloaders=dataloader)
 
 
 @RunIf(min_gpus=2)