diff --git a/pytorch_lightning/utilities/apply_func.py b/pytorch_lightning/utilities/apply_func.py index 2f7425bf3beb0..42d2588195073 100644 --- a/pytorch_lightning/utilities/apply_func.py +++ b/pytorch_lightning/utilities/apply_func.py @@ -25,7 +25,7 @@ from pytorch_lightning.utilities.imports import _TORCHTEXT_AVAILABLE if _TORCHTEXT_AVAILABLE: - from torchtext.data import Batch + from torchtext.legacy.data import Batch else: Batch = type(None)