Skip to content

S
seminar-breakout
Commit 9e436219 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix missing import

parent 41c8446c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tensorpack/dataflow/prefetch.py
View file @ 9e436219
...@@ -15,11 +15,11 @@ from ..utils.concurrency import * ...@@ -15,11 +15,11 @@ from ..utils.concurrency import *
from ..utils.serialize import loads, dumps from ..utils.serialize import loads, dumps
from ..utils import logger, change_env from ..utils import logger, change_env
__all__ = ['PrefetchData', 'BlockParallel']
try: try:
import zmq import zmq
except ImportError: except ImportError:
logger.warn("Error in 'import zmq'. PrefetchDataZMQ won't be available.") logger.warn("Error in 'import zmq'. PrefetchDataZMQ won't be available.")
__all__ = ['PrefetchData', 'BlockParallel']
else: else:
__all__.extend(['PrefetchDataZMQ', 'PrefetchOnGPUs']) __all__.extend(['PrefetchDataZMQ', 'PrefetchOnGPUs'])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment