fix batch typo

Lightning-AI · Jun 29, 2020 · 92d1e75 · 92d1e75
1 parent 593837e
commit 92d1e75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/utilities/apply_func.py b/pytorch_lightning/utilities/apply_func.py
@@ -10,7 +10,7 @@
 if TORCHTEXT_AVAILABLE:
     from torchtext.data import Batch
 else:
-    Batch = None
+    Batch = type(None)
 
 
 def apply_to_collection(data: Any, dtype: Union[type, tuple], function: Callable, *args, **kwargs) -> Any: