expose MultiProcessMapData to docs

53a41ae8 · Yuxin Wu · 5a0fe4e1 · 53a41ae8
Commit 53a41ae8 authored Nov 23, 2017 by Yuxin Wu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

tensorpack/dataflow/prefetch.py tensorpack/dataflow/prefetch.py +2 -1

No files found.
--- a/tensorpack/dataflow/prefetch.py
+++ b/tensorpack/dataflow/prefetch.py
@@ -22,7 +22,8 @@ from ..utils import logger
 from ..utils.gpu import change_gpu

 __all__ = ['PrefetchData', 'PrefetchDataZMQ', 'PrefetchOnGPUs',
-           'ThreadedMapData', 'MultiThreadMapData', 'MultiProcessMapData']
+           'ThreadedMapData', 'MultiThreadMapData',
+           'MultiProcessMapData', 'MultiProcessMapDataZMQ']


 def _repeat_iter(get_itr):