corrected import bug (plotly#2265)

emmanuelle · web-flow · commit ce4fec85e4db · 2020-03-11T11:56:11.000-04:00
diff --git a/packages/python/plotly/codegen/__init__.py b/packages/python/plotly/codegen/__init__.py
@@ -259,9 +259,9 @@ def perform_codegen():
     from distutils.version import LooseVersion
     if LooseVersion(ipywidgets.__version__) >= LooseVersion('7.0.0'):
         from ._figurewidget import FigureWidget
+        __all__.append("FigureWidget")
     del LooseVersion
     del ipywidgets
-    __all__.append("FigureWidget")
 except ImportError:
     pass
 """
diff --git a/packages/python/plotly/plotly/graph_objs/__init__.py b/packages/python/plotly/plotly/graph_objs/__init__.py
@@ -100182,8 +100182,9 @@ def __init__(
 
     if LooseVersion(ipywidgets.__version__) >= LooseVersion("7.0.0"):
         from ._figurewidget import FigureWidget
+
+        __all__.append("FigureWidget")
     del LooseVersion
     del ipywidgets
-    __all__.append("FigureWidget")
 except ImportError:
     pass
