diff --git a/torchtext/__init__.py b/torchtext/__init__.py index 399980150f..6819ecee87 100644 --- a/torchtext/__init__.py +++ b/torchtext/__init__.py @@ -1,9 +1,11 @@ from . import data from . import datasets from . import utils +from . import vocab __version__ = '0.3.0' __all__ = ['data', 'datasets', - 'utils'] + 'utils', + 'vocab'] diff --git a/torchtext/datasets/__init__.py b/torchtext/datasets/__init__.py index b052dad835..1457969809 100644 --- a/torchtext/datasets/__init__.py +++ b/torchtext/datasets/__init__.py @@ -18,7 +18,7 @@ 'WMT14', 'WikiText2', 'WikiText103', - 'PennTreeBank', + 'PennTreebank', 'TREC', 'IMDB', 'SequenceTaggingDataset',