fix build

tensorpack/callbacks/__init__.py

@@ -14,8 +14,9 @@ def _global_import(name):
     lst = p.__all__ if '__all__' in dir(p) else dir(p)
     del globals()[name]
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 _CURR_DIR = os.path.dirname(__file__)

tensorpack/dataflow/__init__.py

@@ -15,8 +15,9 @@ def _global_import(name):
     lst = p.__all__ if '__all__' in dir(p) else dir(p)
     del globals()[name]
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 __SKIP = set(['dftools', 'dataset', 'imgaug'])

tensorpack/dataflow/dataset/__init__.py

@@ -14,8 +14,9 @@ def global_import(name):
     lst = p.__all__ if '__all__' in dir(p) else dir(p)
     del globals()[name]
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 _CURR_DIR = os.path.dirname(__file__)

tensorpack/dataflow/imgaug/__init__.py

@@ -13,8 +13,9 @@ def global_import(name):
     lst = p.__all__ if '__all__' in dir(p) else dir(p)
     del globals()[name]
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 try:

tensorpack/graph_builder/__init__.py

@@ -14,8 +14,9 @@ def global_import(name):
     lst = p.__all__ if '__all__' in dir(p) else []
     del globals()[name]
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 _CURR_DIR = os.path.dirname(__file__)

tensorpack/models/__init__.py

@@ -16,8 +16,9 @@ def _global_import(name):
     lst = p.__all__ if '__all__' in dir(p) else dir(p)
     del globals()[name]
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 _CURR_DIR = os.path.dirname(__file__)

tensorpack/tfutils/__init__.py

@@ -12,8 +12,9 @@ def _global_import(name):
     p = __import__(name, globals(), None, level=1)
     lst = p.__all__ if '__all__' in dir(p) else dir(p)
     for k in lst:
-        globals()[k] = p.__dict__[k]
-        __all__.append(k)
+        if not k.startswith('__'):
+            globals()[k] = p.__dict__[k]
+            __all__.append(k)
 
 
 _TO_IMPORT = set([