try catch for widgets imports

etpinard · etpinard · commit de131c81d482 · 2016-08-25T17:06:38.000-04:00
diff --git a/plotly/widgets/graph_widget.py b/plotly/widgets/graph_widget.py
@@ -7,10 +7,13 @@
 from collections import deque
 from pkg_resources import resource_string
 
+try:
+    from ipywidgets import widgets
+    from traitlets import Unicode
+except ImportError:
+    from IPython.html import widgets
+    from IPython.utils.traitlets import Unicode
 
-# TODO: protected imports?
-# from IPython.html import widgets
-# from IPython.utils.traitlets import Unicode
 from IPython.display import Javascript, display
 
 import plotly.plotly.plotly as py
