def test_pre_made_batches():
    """Check that loader works with pre-made batches."""
    loader = DataLoader(RandomDataset(32, 10), batch_size=None)
    trainer = Trainer(fast_dev_run=1)
    trainer.predict(LoaderTestModel(), loader)
Check that loader works with pre-made batches.
tests/tests_pytorch/trainer/connectors/test_data_connector.py