def METHOD_NAME(tmpdir):
    predict_files = _create_synth_files_dataset(tmpdir)
    datamodule = ObjectDetectionData.from_files(
        predict_files=predict_files, batch_size=1, transform_kwargs={"image_size": 128}
    )
    data = next(iter(datamodule.predict_dataloader()))
    sample = data[0]
    assert sample[DataKeys.INPUT].shape == (128, 128, 3)