fix import

tensorpack · Mar 12, 2020 · 6f0ba59 · 6f0ba59
1 parent b01cd6d
commit 6f0ba59
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/tensorpack/__init__.py b/tensorpack/__init__.py
@@ -19,6 +19,5 @@
     from tensorpack.tfutils import *
 
     from tensorpack.train import *
-    from tensorpack.graph_builder import InputDesc  # kept for BC
     from tensorpack.input_source import *
     from tensorpack.predict import *
diff --git a/tensorpack/graph_builder/__init__.py b/tensorpack/graph_builder/__init__.py
@@ -10,7 +10,7 @@
     from .distributed import *
     from .utils import *
 
-from .model_desc import InputDesc, ModelDesc, ModelDescBase  # kept for BC # noqa
+from .model_desc import ModelDesc, ModelDescBase
 
 from pkgutil import iter_modules
 import os