Commit 2d5b44e6 authored by Yuxin Wu's avatar Yuxin Wu

fix build

parent 5134338e
...@@ -13,7 +13,8 @@ import tqdm ...@@ -13,7 +13,8 @@ import tqdm
import six import six
from six.moves import range from six.moves import range
from ..utils import logger, get_tqdm_kwargs from ..utils import logger
from ..utils.utils import get_tqdm_kwargs
from ..utils.develop import deprecated from ..utils.develop import deprecated
from ..dataflow import DataFlow from ..dataflow import DataFlow
......
...@@ -8,7 +8,8 @@ import tensorflow as tf ...@@ -8,7 +8,8 @@ import tensorflow as tf
from six.moves import zip from six.moves import zip
import tqdm import tqdm
from ..utils import logger, get_tqdm_kwargs from ..utils import logger
from ..utils.utils import get_tqdm_kwargs
from ..utils.naming import GLOBAL_STEP_INCR_OP_NAME from ..utils.naming import GLOBAL_STEP_INCR_OP_NAME
from ..tfutils.common import ( from ..tfutils.common import (
get_op_tensor_name, get_global_step_var, get_op_tensor_name, get_global_step_var,
......
...@@ -10,7 +10,8 @@ from termcolor import colored ...@@ -10,7 +10,8 @@ from termcolor import colored
from collections import deque, defaultdict from collections import deque, defaultdict
from six.moves import range, map from six.moves import range, map
from .base import DataFlow, ProxyDataFlow, RNGDataFlow from .base import DataFlow, ProxyDataFlow, RNGDataFlow
from ..utils import logger, get_tqdm, get_rng from ..utils import logger, get_rng
from ..utils.utils import get_tqdm
from ..utils.develop import log_deprecated from ..utils.develop import log_deprecated
__all__ = ['TestDataSpeed', 'PrintData', 'BatchData', 'BatchDataByShape', 'FixedSizeData', 'MapData', __all__ = ['TestDataSpeed', 'PrintData', 'BatchData', 'BatchDataByShape', 'FixedSizeData', 'MapData',
......
...@@ -7,7 +7,8 @@ import multiprocessing as mp ...@@ -7,7 +7,8 @@ import multiprocessing as mp
from six.moves import range from six.moves import range
from .base import DataFlow from .base import DataFlow
from ..utils import get_tqdm, logger from ..utils import logger
from ..utils.utils import get_tqdm
from ..utils.concurrency import DIE from ..utils.concurrency import DIE
from ..utils.serialize import dumps from ..utils.serialize import dumps
......
...@@ -7,7 +7,8 @@ import six ...@@ -7,7 +7,8 @@ import six
from six.moves import range from six.moves import range
import os import os
from ..utils import logger, get_tqdm from ..utils import logger
from ..utils.utils import get_tqdm
from ..utils.timer import timed_operation from ..utils.timer import timed_operation
from ..utils.loadcaffe import get_caffe_pb from ..utils.loadcaffe import get_caffe_pb
from ..utils.serialize import loads from ..utils.serialize import loads
......
...@@ -15,7 +15,7 @@ __all__ = ['change_env', ...@@ -15,7 +15,7 @@ __all__ = ['change_env',
'get_rng', 'get_rng',
'fix_rng_seed', 'fix_rng_seed',
# 'get_tqdm_kwargs', # 'get_tqdm_kwargs',
# 'get_tqdm', 'get_tqdm',
'execute_only_once', 'execute_only_once',
] ]
...@@ -122,6 +122,6 @@ def get_tqdm_kwargs(**kwargs): ...@@ -122,6 +122,6 @@ def get_tqdm_kwargs(**kwargs):
def get_tqdm(**kwargs): def get_tqdm(**kwargs):
""" Similar to :func:`get_tqdm_kwargs`, but returns the tqdm object """ Similar to :func:`get_tqdm_kwargs`,
directly. """ but returns the tqdm object directly. """
return tqdm(**get_tqdm_kwargs(**kwargs)) return tqdm(**get_tqdm_kwargs(**kwargs))
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