import MultiTaskDataLoader to data_loaders/__init__.py (#4885)

allenai · Dec 23, 2020 · fbab0bd · fbab0bd
1 parent d1cc146
commit fbab0bd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/allennlp/data/data_loaders/__init__.py b/allennlp/data/data_loaders/__init__.py
@@ -1,5 +1,7 @@
 from allennlp.data.data_loaders.data_loader import DataLoader, TensorDict, allennlp_collate
 from allennlp.data.data_loaders.multi_process_data_loader import MultiProcessDataLoader, WorkerError
+from allennlp.data.data_loaders.multitask_data_loader import MultiTaskDataLoader
+
 from allennlp.data.data_loaders.pytorch_data_loader import (
     PyTorchDataLoader,
     allennlp_worker_init_fn,