Commit 0038c29e authored by Yuxin Wu's avatar Yuxin Wu

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

parent 326f335e
......@@ -19,8 +19,12 @@ def _global_import(name):
__SKIP = set(['dftools', 'dataset', 'imgaug'])
_CURR_DIR = os.path.dirname(__file__)
for _, module_name, __ in iter_modules(
[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 \
module_name not in __SKIP:
_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