Add missing imports

Summary: Pull Request resolved: https://github.com/fairinternal/fairseq-py/pull/579

Differential Revision: D15352058

Pulled By: myleott

fbshipit-source-id: cebef02edcfcb203ef2e32c64f7f28e08c4e46b0

fairseq/data/__init__.py

@@ -6,12 +6,17 @@
 # can be found in the PATENTS file in the same directory.
 
 from .dictionary import Dictionary, TruncatedDictionary
+from .masked_lm_dictionary import BertDictionary, MaskedLMDictionary
+
 from .fairseq_dataset import FairseqDataset
+
 from .backtranslation_dataset import BacktranslationDataset
+from .block_pair_dataset import BlockPairDataset
 from .concat_dataset import ConcatDataset
 from .indexed_dataset import IndexedCachedDataset, IndexedDataset, IndexedRawTextDataset, MMapIndexedDataset
 from .language_pair_dataset import LanguagePairDataset
 from .lm_context_window_dataset import LMContextWindowDataset
+from .masked_lm_dataset import MaskedLMDataset
 from .monolingual_dataset import MonolingualDataset
 from .noising import NoisingDataset
 from .round_robin_zip_datasets import RoundRobinZipDatasets
@@ -28,6 +33,8 @@ from .iterators import (
 
 __all__ = [
     'BacktranslationDataset',
+    'BertDictionary',
+    'BlockPairDataset',
     'ConcatDataset',
     'CountingIterator',
     'Dictionary',
@@ -39,6 +46,8 @@ __all__ = [
     'IndexedRawTextDataset',
     'LanguagePairDataset',
     'LMContextWindowDataset',
+    'MaskedLMDataset',
+    'MaskedLMDictionary',
     'MMapIndexedDataset',
     'MonolingualDataset',
     'NoisingDataset',