Import internal references as relative imports

jtpio · Sep 6, 2022 · db41eab · db41eab
1 parent 25acec5
commit db41eab
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/python/ipywidgets/ipywidgets/embed.py b/python/ipywidgets/ipywidgets/embed.py
@@ -12,8 +12,7 @@
 
 import json
 import re
-import ipywidgets.widgets.widget
-from .widgets import Widget, DOMWidget
+from .widgets import Widget, DOMWidget, widget as widget_module
 from .widgets.widget_link import Link
 from .widgets.docutils import doc_subst
 from ._version import __html_manager_version__
@@ -130,7 +129,7 @@ def _get_recursive_state(widget, store=None, drop_defaults=False):
 
 def add_resolved_links(store, drop_defaults):
     """Adds the state of any link models between two models in store"""
-    for widget_id, widget in ipywidgets.widgets.widget._instances.items(): # go over all widgets
+    for widget_id, widget in widget_module._instances.items(): # go over all widgets
         if isinstance(widget, Link) and widget_id not in store:
             if widget.source[0].model_id in store and widget.target[0].model_id in store:
                 store[widget.model_id] = widget._get_embed_state(drop_defaults=drop_defaults)
@@ -208,7 +207,7 @@ def embed_data(views, drop_defaults=True, state=None):
         view_specs: a list of widget view specs
     """
     if views is None:
-        views = [w for w in ipywidgets.widgets.widget._instances.values() if isinstance(w, DOMWidget)]
+        views = [w for w in widget_module._instances.values() if isinstance(w, DOMWidget)]
     else:
         try:
             views[0]

diff --git a/python/ipywidgets/ipywidgets/tests/test_embed.py b/python/ipywidgets/ipywidgets/tests/test_embed.py
@@ -9,8 +9,7 @@
 
 import traitlets
 
-import ipywidgets.widgets.widget
-from ..widgets import IntSlider, IntText, Text, Widget, jslink, HBox, widget_serialization
+from ..widgets import IntSlider, IntText, Text, Widget, jslink, HBox, widget_serialization, widget as widget_module
 from ..embed import embed_data, embed_snippet, embed_minimal_html, dependency_state
 
 
@@ -30,7 +29,7 @@ class CaseWidget(Widget):
 class TestEmbed:
 
     def teardown(self):
-        for w in tuple(ipywidgets.widgets.widget._instances.values()):
+        for w in tuple(widget_module._instances.values()):
             w.close()
 
     def test_embed_data_simple(self):