fix batch typo

2020-06-29 06:54:21 -04:00 · 2020-06-29 06:54:21 -04:00 · 92d1e75b26
parent 593837e1da
commit 92d1e75b26
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/utilities/apply_func.py
+++ b/pytorch_lightning/utilities/apply_func.py
@ -10,7 +10,7 @@ TORCHTEXT_AVAILABLE = importlib.util.find_spec("torchtext") is not None
 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: