From 92d1e75b2638a493d9d21ed5fe00a22093888285 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Mon, 29 Jun 2020 06:54:21 -0400 Subject: [PATCH] fix batch typo --- pytorch_lightning/utilities/apply_func.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch_lightning/utilities/apply_func.py b/pytorch_lightning/utilities/apply_func.py index f81a382041..5846214636 100644 --- 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: