use iter_modules to import everyfile. fix #114.

tensorpack/RL/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 import os.path
 
@@ -18,7 +18,7 @@ def _global_import(name):
         __all__.append(k)
 
 
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [os.path.dirname(__file__)]):
     if not module_name.startswith('_'):
         _global_import(module_name)

tensorpack/callbacks/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 
 
@@ -19,7 +19,7 @@ def _global_import(name):
 
 
 _CURR_DIR = os.path.dirname(__file__)
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [_CURR_DIR]):
     srcpath = os.path.join(_CURR_DIR, module_name + '.py')
     if not os.path.isfile(srcpath):

tensorpack/dataflow/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 import os.path
 
@@ -22,7 +22,7 @@ def _global_import(name):
 
 
 __SKIP = ['dftools', 'dataset', 'imgaug']
-for _, module_name, _ in walk_packages(
+for _, module_name, __ in iter_modules(
         [os.path.dirname(__file__)]):
     if not module_name.startswith('_') and \
             module_name not in __SKIP:

tensorpack/dataflow/dataset/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 import os.path
 
@@ -19,7 +19,7 @@ def global_import(name):
 
 
 _CURR_DIR = os.path.dirname(__file__)
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [_CURR_DIR]):
     srcpath = os.path.join(_CURR_DIR, module_name + '.py')
     if not os.path.isfile(srcpath):

tensorpack/dataflow/imgaug/__init__.py

@@ -3,7 +3,7 @@
 # Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
 import os
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 
 __all__ = []
 
@@ -17,7 +17,7 @@ def global_import(name):
         __all__.append(k)
 
 
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [os.path.dirname(__file__)]):
     if not module_name.startswith('_'):
         global_import(module_name)

tensorpack/models/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 from types import ModuleType
 import six
 import os
@@ -23,7 +23,7 @@ def _global_import(name):
         __all__.append(k)
 
 
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [os.path.dirname(__file__)]):
     if not module_name.startswith('_'):
         _global_import(module_name)

tensorpack/predict/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 import os.path
 
@@ -19,7 +19,7 @@ def global_import(name):
 
 
 _CURR_DIR = os.path.dirname(__file__)
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [_CURR_DIR]):
     srcpath = os.path.join(_CURR_DIR, module_name + '.py')
     if not os.path.isfile(srcpath):

tensorpack/tfutils/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 
 __all__ = []
@@ -25,7 +25,7 @@ _TO_IMPORT = set([
     ])
 
 _CURR_DIR = os.path.dirname(__file__)
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [_CURR_DIR]):
     srcpath = os.path.join(_CURR_DIR, module_name + '.py')
     if not os.path.isfile(srcpath):

tensorpack/train/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 import os.path
 
@@ -19,7 +19,7 @@ def global_import(name):
 
 
 _CURR_DIR = os.path.dirname(__file__)
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [_CURR_DIR]):
     srcpath = os.path.join(_CURR_DIR, module_name + '.py')
     if not os.path.isfile(srcpath):

tensorpack/utils/__init__.py

@@ -2,7 +2,7 @@
 #  File: __init__.py
 #  Author: Yuxin Wu <ppwwyyxx@gmail.com>
 
-from pkgutil import walk_packages
+from pkgutil import iter_modules
 import os
 
 """
@@ -28,7 +28,7 @@ _TO_IMPORT = set([
 ])
 
 _CURR_DIR = os.path.dirname(__file__)
-for _, module_name, _ in walk_packages(
+for _, module_name, _ in iter_modules(
         [_CURR_DIR]):
     srcpath = os.path.join(_CURR_DIR, module_name + '.py')
     if not os.path.isfile(srcpath):