Merge 58b5a59 into e7298b5

Author: dbonner

pytorch_lightning/utilities/apply_func.py

@@ -23,9 +23,13 @@
 
 from pytorch_lightning.utilities.exceptions import MisconfigurationException
 from pytorch_lightning.utilities.imports import _TORCHTEXT_AVAILABLE
+from pytorch_lightning.utilities.imports import _module_available
 
 if _TORCHTEXT_AVAILABLE:
-    from torchtext.data import Batch
+    if _module_available("torchtext.legacy.data"):
+        from torchtext.legacy.data import Batch
+    else:
+        from torchtext.data import Batch
 else:
     Batch = type(None)