Skip to content

S
seminar-breakout
Commit 0038c29e authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

check deleted modules in dataflow/ to fix 'git pull' users

parent 326f335e
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tensorpack/dataflow/__init__.py
View file @ 0038c29e
...@@ -19,8 +19,12 @@ def _global_import(name): ...@@ -19,8 +19,12 @@ def _global_import(name):
__SKIP = set(['dftools', 'dataset', 'imgaug']) __SKIP = set(['dftools', 'dataset', 'imgaug'])
_CURR_DIR = os.path.dirname(__file__)
for _, module_name, __ in iter_modules( for _, module_name, __ in iter_modules(
[os.path.dirname(__file__)]): [os.path.dirname(__file__)]):
srcpath = os.path.join(_CURR_DIR, module_name + '.py')
if not os.path.isfile(srcpath):
continue
if not module_name.startswith('_') and \ if not module_name.startswith('_') and \
module_name not in __SKIP: module_name not in __SKIP:
_global_import(module_name) _global_import(module_name)
......
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