From 1562213a7e4f0688a541140268c79f66593513da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Adrian=20W=C3=A4lchli?= <adrian.waelchli@inf.unibe.ch>
Date: Sun, 28 Jun 2020 21:02:32 +0200
Subject: [PATCH] fix when torchtext not installed

---
 pytorch_lightning/utilities/apply_func.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pytorch_lightning/utilities/apply_func.py b/pytorch_lightning/utilities/apply_func.py
index 2003bca6fd2e1..f81a38204184d 100644
--- a/pytorch_lightning/utilities/apply_func.py
+++ b/pytorch_lightning/utilities/apply_func.py
@@ -9,6 +9,8 @@
 TORCHTEXT_AVAILABLE = importlib.util.find_spec("torchtext") is not None
 if TORCHTEXT_AVAILABLE:
     from torchtext.data import Batch
+else:
+    Batch = None
 
 
 def apply_to_collection(data: Any, dtype: Union[type, tuple], function: Callable, *args, **kwargs) -> Any: