Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix batch typo

This commit is contained in:
William Falcon 2020-06-29 06:54:21 -04:00
parent 593837e1da
commit 92d1e75b26
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/utilities/apply_func.py
View File

@ -10,7 +10,7 @@ TORCHTEXT_AVAILABLE = importlib.util.find_spec("torchtext") is not None
if TORCHTEXT_AVAILABLE: if TORCHTEXT_AVAILABLE:
from torchtext.data import Batch from torchtext.data import Batch
else: else:
Batch = None Batch = type(None)
def apply_to_collection(data: Any, dtype: Union[type, tuple], function: Callable, *args, **kwargs) -> Any: def apply_to_collection(data: Any, dtype: Union[type, tuple], function: Callable, *args, **kwargs) -> Any: