Skip to content

S
seminar-breakout
Commit 4efa7108 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix import #92

parent e0190688
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 2 deletions
tensorpack/tfutils/__init__.py
View file @ 4efa7108
...@@ -33,5 +33,7 @@ for _, module_name, _ in walk_packages( ...@@ -33,5 +33,7 @@ for _, module_name, _ in walk_packages(
if module_name.startswith('_'): if module_name.startswith('_'):
continue continue
if module_name in _TO_IMPORT: if module_name in _TO_IMPORT:
_global_import(module_name) _global_import(module_name) # import the content to tfutils.*
else:
__all__.append(module_name) # import the module separately
__all__.extend(['sessinit', 'gradproc']) __all__.extend(['sessinit', 'gradproc'])
tensorpack/utils/__init__.py
View file @ 4efa7108
...@@ -37,4 +37,5 @@ for _, module_name, _ in walk_packages( ...@@ -37,4 +37,5 @@ for _, module_name, _ in walk_packages(
continue continue
if module_name in _TO_IMPORT: if module_name in _TO_IMPORT:
_global_import(module_name) _global_import(module_name)
else:
__all__.append(module_name) __all__.append(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