pyglove 0.4.5.dev20240319__py3-none-any.whl → 0.4.5.dev202501132210__py3-none-any.whl

pyglove/core/symbolic/diff.py

@@ -13,17 +13,18 @@
 # limitations under the License.
 """Symbolic differences."""
 
-from typing import Any, Callable, Tuple, Union
+from typing import Any, Callable, Optional, Sequence, Tuple, Union
 
-from pyglove.core import object_utils
 from pyglove.core import typing as pg_typing
+from pyglove.core import utils
 from pyglove.core.symbolic import base
 from pyglove.core.symbolic import list as pg_list
 from pyglove.core.symbolic import object as pg_object
 from pyglove.core.symbolic.pure_symbolic import PureSymbolic
+from pyglove.core.views.html import tree_view
 
 
-class Diff(PureSymbolic, pg_object.Object):
+class Diff(PureSymbolic, pg_object.Object, tree_view.HtmlTreeView.Extension):
   """A value diff between two objects: a 'left' object and a 'right' object.
 
   If one of them is missing, it may be represented by pg.Diff.MISSING
@@ -111,8 +112,7 @@ class Diff(PureSymbolic, pg_object.Object):
         return 'No diff'
       # When there is no diff, but the same value needs to be displayed
       # we simply return the value.
-      return object_utils.format(
-          self.value, compact, verbose, root_indent, **kwargs)
+      return utils.format(self.value, compact, verbose, root_indent, **kwargs)
     if self.is_leaf:
       exclude_keys = kwargs.pop('exclude_keys', None)
       exclude_keys = exclude_keys or set()
@@ -128,7 +128,7 @@ class Diff(PureSymbolic, pg_object.Object):
             verbose=verbose,
             root_indent=root_indent,
             cls_name='',
-            bracket_type=object_utils.BracketType.SQUARE,
+            bracket_type=utils.BracketType.SQUARE,
             **kwargs,
         )
       if self.left is self.right:
@@ -140,10 +140,198 @@ class Diff(PureSymbolic, pg_object.Object):
           verbose=verbose,
           root_indent=root_indent,
           cls_name=cls_name,
-          bracket_type=object_utils.BracketType.ROUND,
+          bracket_type=utils.BracketType.ROUND,
           **kwargs,
       )
 
+  # pytype: disable=annotation-type-mismatch
+  def _html_tree_view_summary(
+      self,
+      *,
+      view: tree_view.HtmlTreeView,
+      css_classes: Optional[Sequence[str]] = None,
+      title: Optional[str] = None,
+      max_summary_len_for_str: int = 80,
+      **kwargs,
+    ) -> Optional[tree_view.Html]:
+    # pytype: enable=annotation-type-mismatch
+    if not bool(self):
+      v = self.value
+      if (isinstance(v, (int, float, bool, type(None)))
+          or (isinstance(v, str) and len(v) <= max_summary_len_for_str)):
+        return None
+
+      if v == Diff.MISSING:
+        return view.summary(
+            self,
+            title=title or 'Diff',
+            css_classes=css_classes,
+            max_summary_len_for_str=max_summary_len_for_str,
+            **kwargs
+        )
+      return view.summary(
+          self.value,
+          title=title,
+          css_classes=css_classes,
+          max_summary_len_for_str=max_summary_len_for_str,
+          **kwargs
+      )
+    elif self.is_leaf:
+      return None
+    else:
+      assert isinstance(self.left, type), self.left
+      assert isinstance(self.right, type), self.right
+      if self.left is self.right:
+        diff_title = self.left.__name__
+      else:
+        diff_title = f'{self.left.__name__} | {self.right.__name__}'
+      return view.summary(
+          self,
+          title=title or diff_title,
+          css_classes=css_classes,
+          max_summary_len_for_str=max_summary_len_for_str,
+          **kwargs
+      )
+
+  def _html_tree_view_content(
+      self,
+      *,
+      view: tree_view.HtmlTreeView,
+      parent: Any = None,
+      root_path: Optional[utils.KeyPath] = None,
+      css_classes: Optional[Sequence[str]] = None,
+      **kwargs,
+  ) -> tree_view.Html:
+    root_path = root_path or utils.KeyPath()
+    if not bool(self):
+      if self.value == Diff.MISSING:
+        root = tree_view.Html.element(
+            'span',
+            # CSS class already defined in HtmlTreeView.
+            css_classes=['diff-empty']
+        )
+      else:
+        # When there is no diff, but the same value needs to be displayed
+        # we simply return the value.
+        root = view.content(
+            self.value, parent=parent, root_path=root_path,
+            css_classes=['no-diff', view.css_class_name(self.value)],
+            **kwargs,
+        )
+    elif self.is_leaf:
+      root = tree_view.Html.element(
+          'div',
+          [
+              view.render(   # pylint: disable=g-long-ternary
+                  self.left, root_path=root_path + 'left',
+                  css_classes=['diff-left'], **kwargs
+              ) if self.left != Diff.MISSING else None,
+              view.render(   # pylint: disable=g-long-ternary
+                  self.right, root_path=root_path + 'right',
+                  css_classes=['diff-right'], **kwargs
+              ) if self.right != Diff.MISSING else None,
+          ],
+          css_classes=['diff-value']
+      )
+    else:
+      assert isinstance(self.left, type)
+      assert isinstance(self.right, type)
+
+      if issubclass(self.left, list):
+        key_fn = int
+      else:
+        key_fn = lambda k: k
+
+      s = tree_view.Html()
+      for k, v in self.children.items():
+        k = key_fn(k)
+        child_path = root_path + k
+        has_diff = bool(v)
+        s.write('<tr><td>')
+        # Print Key.
+        s.write(
+            view.object_key(
+                child_path, value=v, parent=self,
+                css_classes=None if has_diff else ['no-diff'],
+            ),
+        )
+        # Print Value.
+        s.write(
+            '</td><td>',
+            view.render(
+                v, root_path=child_path, **kwargs
+            ),
+            '</td></tr>'
+        )
+      root = tree_view.Html.element(
+          'div',
+          [
+              '<table>', s, '</table>'
+          ],
+          css_classes=[
+              'complex-value', view.css_class_name(self.left), css_classes,
+          ]
+      )
+    return root
+
+  def _html_tree_view_config(self) -> dict[str, Any]:
+    return tree_view.HtmlTreeView.get_kwargs(
+        super()._html_tree_view_config(),
+        dict(
+            css_classes=[
+                'has-diff' if bool(self) else 'no-diff'
+            ]
+        )
+    )
+
+  @classmethod
+  def _html_tree_view_css_styles(cls) -> list[str]:
+    return super()._html_tree_view_css_styles() + [
+        """
+        /* Diff styles. */
+        .has-diff.summary-title::after {
+          content: ' (diff)';
+          color: #aaa;
+        }
+        .has-diff.summary-title {
+          background-color: yellow;
+        }
+        .no-diff.summary-title::after {
+          content: ' (no diff)';
+          font-style: italic;
+          font-weight: normal;
+          color: #aaa;
+        }
+        .no-diff {
+          opacity: 0.6;
+        }
+        .no-diff.simple_value::after {
+          content: '(no diff)';
+          margin-left: 0.5em;
+          color: #aaa;
+          font-style: italic;
+        }
+        .diff-empty::before {
+          content: '(empty)';
+          font-style: italic;
+          margin-left: 0.5em;
+          color: #aaa;
+        }
+        .diff-left.summary-title::before, .diff-left.simple-value::before{
+          content: '🇱';
+        }
+        .diff-left {
+          background-color: #ffcccc;
+        }
+        .diff-right.summary-title::before, .diff-right.simple-value::before{
+          content: '🇷';
+        }
+        .diff-right {
+          background-color: #ccffcc;
+        }
+        """
+    ]
+
 
 # NOTE(daiyip): we add the symbolic attribute to Diff after its declaration
 # since we need to access Diff.MISSING as the default value for `left` and
@@ -164,7 +352,8 @@ def diff(
     right: Any,
     flatten: bool = False,
     collapse: Union[bool, str, Callable[[Any, Any], bool]] = 'same_type',
-    mode: str = 'diff') -> object_utils.Nestable[Diff]:
+    mode: str = 'diff',
+) -> utils.Nestable[Diff]:
   """Inspect the symbolic diff between two objects.
 
   For example::
@@ -290,7 +479,7 @@ def diff(
       assert isinstance(container, base.Symbolic)
       return container.sym_hasattr, container.sym_getattr, container.sym_items
 
-  def _diff(x, y) -> Tuple[object_utils.Nestable[Diff], bool]:
+  def _diff(x, y) -> Tuple[utils.Nestable[Diff], bool]:
     if x is y or x == y:
       return (Diff(x, y), False)
     if not _should_collapse(x, y):
@@ -344,5 +533,5 @@ def diff(
   if not has_diff and mode == 'diff':
     diff_value = Diff()
   if flatten:
-    diff_value = object_utils.flatten(diff_value)
+    diff_value = utils.flatten(diff_value)
   return diff_value

pyglove/core/symbolic/diff_test.py

@@ -13,6 +13,8 @@
 # limitations under the License.
 """Tests for pyglove.diff."""
 
+import inspect
+from typing import Any
 import unittest
 
 from pyglove.core import typing as pg_typing
@@ -337,6 +339,230 @@ class DiffTest(unittest.TestCase):
             '_type': Diff(B, C),
         })
 
+  def test_to_html(self):
+
+    class Foo(Object):
+      x: Any
+      y: Any
+
+    class Bar(Foo):
+      pass
+
+    def assert_style(html, expected):
+      expected = inspect.cleandoc(expected).strip()
+      actual = html.style_section.strip()
+      if expected not in actual:
+        print(actual)
+      self.assertIn(expected, actual)
+
+    def assert_content(html, expected):
+      expected = inspect.cleandoc(expected).strip()
+      actual = html.content.strip()
+      if actual != expected:
+        print(actual)
+      self.assertEqual(actual.strip(), expected)
+
+    assert_style(
+        pg_diff(1, 1).to_html(),
+        """
+        /* Diff styles. */
+        .has-diff.summary-title::after {
+          content: ' (diff)';
+          color: #aaa;
+        }
+        .has-diff.summary-title {
+          background-color: yellow;
+        }
+        .no-diff.summary-title::after {
+          content: ' (no diff)';
+          font-style: italic;
+          font-weight: normal;
+          color: #aaa;
+        }
+        .no-diff {
+          opacity: 0.6;
+        }
+        .no-diff.simple_value::after {
+          content: '(no diff)';
+          margin-left: 0.5em;
+          color: #aaa;
+          font-style: italic;
+        }
+        .diff-empty::before {
+          content: '(empty)';
+          font-style: italic;
+          margin-left: 0.5em;
+          color: #aaa;
+        }
+        .diff-left.summary-title::before, .diff-left.simple-value::before{
+          content: '🇱';
+        }
+        .diff-left {
+          background-color: #ffcccc;
+        }
+        .diff-right.summary-title::before, .diff-right.simple-value::before{
+          content: '🇷';
+        }
+        .diff-right {
+          background-color: #ccffcc;
+        }
+        """
+    )
+
+    # No diff leaf value (diff only)
+    assert_content(
+        pg_diff(1, 1).to_html(),
+        """
+        <details open class="pyglove diff no-diff"><summary><div class="summary-title no-diff">Diff</div><span class="tooltip no-diff">No diff</span></summary><span class="diff-empty"></span></details>
+        """
+    )
+
+    # No diff leaf value (keep both)
+    assert_content(
+        pg_diff(1, 1, mode='both').to_html(),
+        """
+        <span class="simple-value int no-diff">1</span>
+        """
+    )
+    # No diff complex value (diff only)
+    assert_content(
+        pg_diff(
+            Foo(x=1, y=[Foo(x=2, y=3)]),
+            Foo(x=1, y=[Foo(x=2, y=3)])
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff no-diff"><summary><div class="summary-title no-diff">Diff</div></summary><span class="diff-empty"></span></details>
+        """
+    )
+    # No diff complex value.
+    assert_content(
+        pg_diff(
+            Foo(x=1, y=[Foo(x=2, y=3)]),
+            Foo(x=1, y=[Foo(x=2, y=3)]), mode='both'
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff no-diff"><summary><div class="summary-title no-diff">Foo(...)</div></summary><div class="complex-value foo no-diff"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details><details class="pyglove list"><summary><div class="summary-name">y</div><div class="summary-title">List(...)</div></summary><div class="complex-value list"><table><tr><td><span class="object-key int">0</span></td><td><details class="pyglove foo"><summary><div class="summary-title">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details></div></details></td></tr></table></div></details></div></details>
+        """
+    )
+
+    # Diff on simple value.
+    assert_content(
+        pg_diff(1, 2).to_html(),
+        """
+        <div class="diff-value"><span class="simple-value int diff-left">1</span><span class="simple-value int diff-right">2</span></div>
+        """
+    )
+
+    # Diff on list.
+    assert_content(
+        pg_diff([0, 1, 2], [0, 2], mode='both').to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">List</div></summary><div class="complex-value list-class"><table><tr><td><span class="object-key int no-diff">0</span><span class="tooltip">[0]</span></td><td><span class="simple-value int no-diff">0</span></td></tr><tr><td><span class="object-key int">1</span><span class="tooltip">[1]</span></td><td><div class="diff-value"><span class="simple-value int diff-left">1</span><span class="simple-value int diff-right">2</span></div></td></tr><tr><td><span class="object-key int">2</span><span class="tooltip">[2]</span></td><td><div class="diff-value"><span class="simple-value int diff-left">2</span></div></td></tr></table></div></details>
+        """
+    )
+
+    # Diff on dict.
+    assert_content(
+        pg_diff(dict(x=1, y=2, z=3), dict(x=1, y=3, w=4), mode='both').to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">dict</div></summary><div class="complex-value dict-class"><table><tr><td><span class="object-key str no-diff">x</span><span class="tooltip">x</span></td><td><span class="simple-value int no-diff">1</span></td></tr><tr><td><span class="object-key str">y</span><span class="tooltip">y</span></td><td><div class="diff-value"><span class="simple-value int diff-left">2</span><span class="simple-value int diff-right">3</span></div></td></tr><tr><td><span class="object-key str">z</span><span class="tooltip">z</span></td><td><div class="diff-value"><span class="simple-value int diff-left">3</span></div></td></tr><tr><td><span class="object-key str">w</span><span class="tooltip">w</span></td><td><div class="diff-value"><span class="simple-value int diff-right">4</span></div></td></tr></table></div></details>
+        """
+    )
+
+    # Diff on symbolic objects of the same type.
+    assert_content(
+        pg_diff(
+            Foo(x=2, y=Foo(x=3, y=3)),
+            Foo(x=1, y=Foo(x=2, y=3)),
+            mode='both',
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">Foo</div></summary><div class="complex-value foo-class"><table><tr><td><span class="object-key str">x</span><span class="tooltip">x</span></td><td><div class="diff-value"><span class="simple-value int diff-left">2</span><span class="simple-value int diff-right">1</span></div></td></tr><tr><td><span class="object-key str">y</span><span class="tooltip">y</span></td><td><details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">Foo</div></summary><div class="complex-value foo-class"><table><tr><td><span class="object-key str">x</span><span class="tooltip">y.x</span></td><td><div class="diff-value"><span class="simple-value int diff-left">3</span><span class="simple-value int diff-right">2</span></div></td></tr><tr><td><span class="object-key str no-diff">y</span><span class="tooltip">y.y</span></td><td><span class="simple-value int no-diff">3</span></td></tr></table></div></details></td></tr></table></div></details>
+        """
+    )
+
+    # Diff on symbolic objects of different types.
+    assert_content(
+        pg_diff(
+            Foo(x=2, y=Foo(x=3, y=3)),
+            Bar(x=2, y=Foo(x=3, y=3)),
+            mode='both',
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <div class="diff-value"><details open class="pyglove foo diff-left"><summary><div class="summary-title diff-left">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details><details class="pyglove foo"><summary><div class="summary-name">y</div><div class="summary-title">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details></div></details></div></details><details open class="pyglove bar diff-right"><summary><div class="summary-title diff-right">Bar(...)</div></summary><div class="complex-value bar"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details><details class="pyglove foo"><summary><div class="summary-name">y</div><div class="summary-title">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details></div></details></div></details></div>
+        """
+    )
+
+    # Different types but same values, with value collapsing.
+    assert_content(
+        pg_diff(
+            Foo(x=2, y=Foo(x=3, y=3)),
+            Bar(x=2, y=Bar(x=3, y=3)),
+            mode='both', collapse=True,
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">Foo | Bar</div></summary><div class="complex-value foo-class"><table><tr><td><span class="object-key str no-diff">x</span><span class="tooltip">x</span></td><td><span class="simple-value int no-diff">2</span></td></tr><tr><td><span class="object-key str">y</span><span class="tooltip">y</span></td><td><details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">Foo | Bar</div></summary><div class="complex-value foo-class"><table><tr><td><span class="object-key str no-diff">x</span><span class="tooltip">y.x</span></td><td><span class="simple-value int no-diff">3</span></td></tr><tr><td><span class="object-key str no-diff">y</span><span class="tooltip">y.y</span></td><td><span class="simple-value int no-diff">3</span></td></tr></table></div></details></td></tr></table></div></details>
+        """
+    )
+
+    # Different types and different values, with value collapsing.
+    assert_content(
+        pg_diff(
+            Foo(x=2, y=Foo(x=3, y=3)),
+            Bar(x=3, y=Bar(x=2, y=3)),
+            mode='both', collapse=True,
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+        ),
+        """
+        <details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">Foo | Bar</div></summary><div class="complex-value foo-class"><table><tr><td><span class="object-key str">x</span><span class="tooltip">x</span></td><td><div class="diff-value"><span class="simple-value int diff-left">2</span><span class="simple-value int diff-right">3</span></div></td></tr><tr><td><span class="object-key str">y</span><span class="tooltip">y</span></td><td><details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">Foo | Bar</div></summary><div class="complex-value foo-class"><table><tr><td><span class="object-key str">x</span><span class="tooltip">y.x</span></td><td><div class="diff-value"><span class="simple-value int diff-left">3</span><span class="simple-value int diff-right">2</span></div></td></tr><tr><td><span class="object-key str no-diff">y</span><span class="tooltip">y.y</span></td><td><span class="simple-value int no-diff">3</span></td></tr></table></div></details></td></tr></table></div></details>
+        """
+    )
+
+    # Diff with uncollapsing UI.
+    assert_content(
+        pg_diff(
+            [
+                Foo(1, 2), Foo(1, 2), None,
+                Foo(x=1, y=Foo(2, 3)), [dict(x=1, y=2)]
+            ],
+            [
+                Foo(1, 2), Bar(1, 2), dict(x=1),
+                [1, 2], Foo(x=2, y=Foo(2, 4)), [dict(x=3)]
+            ],
+            mode='both'
+        ).to_html(
+            enable_summary_tooltip=False,
+            enable_key_tooltip=False,
+            uncollapse=['[0]', '[1].right', '[3].left.y'],
+        ),
+        """
+        <details open class="pyglove diff has-diff"><summary><div class="summary-title has-diff">List</div></summary><div class="complex-value list-class"><table><tr><td><span class="object-key int no-diff">0</span><span class="tooltip">[0]</span></td><td><details open class="pyglove diff no-diff"><summary><div class="summary-title no-diff">Foo(...)</div></summary><div class="complex-value foo no-diff"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details></div></details></td></tr><tr><td><span class="object-key int">1</span><span class="tooltip">[1]</span></td><td><div class="diff-value"><details open class="pyglove foo diff-left"><summary><div class="summary-title diff-left">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details></div></details><details open class="pyglove bar diff-right"><summary><div class="summary-title diff-right">Bar(...)</div></summary><div class="complex-value bar"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details></div></details></div></td></tr><tr><td><span class="object-key int">2</span><span class="tooltip">[2]</span></td><td><div class="diff-value"><span class="simple-value none-type diff-left">None</span><details open class="pyglove dict diff-right"><summary><div class="summary-title diff-right">Dict(...)</div></summary><div class="complex-value dict"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details></div></details></div></td></tr><tr><td><span class="object-key int">3</span><span class="tooltip">[3]</span></td><td><div class="diff-value"><details open class="pyglove foo diff-left"><summary><div class="summary-title diff-left">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details><details open class="pyglove foo"><summary><div class="summary-name">y</div><div class="summary-title">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details></div></details></div></details><details open class="pyglove list diff-right"><summary><div class="summary-title diff-right">List(...)</div></summary><div class="complex-value list"><table><tr><td><span class="object-key int">0</span></td><td><span class="simple-value int">1</span></td></tr><tr><td><span class="object-key int">1</span></td><td><span class="simple-value int">2</span></td></tr></table></div></details></div></td></tr><tr><td><span class="object-key int">4</span><span class="tooltip">[4]</span></td><td><div class="diff-value"><details open class="pyglove list diff-left"><summary><div class="summary-title diff-left">List(...)</div></summary><div class="complex-value list"><table><tr><td><span class="object-key int">0</span></td><td><details class="pyglove dict"><summary><div class="summary-title">Dict(...)</div></summary><div class="complex-value dict"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">1</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details></div></details></td></tr></table></div></details><details open class="pyglove foo diff-right"><summary><div class="summary-title diff-right">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details><details class="pyglove foo"><summary><div class="summary-name">y</div><div class="summary-title">Foo(...)</div></summary><div class="complex-value foo"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">2</span></details><details open class="pyglove int"><summary><div class="summary-name">y</div><div class="summary-title">int</div></summary><span class="simple-value int">4</span></details></div></details></div></details></div></td></tr><tr><td><span class="object-key int">5</span><span class="tooltip">[5]</span></td><td><div class="diff-value"><details open class="pyglove list diff-right"><summary><div class="summary-title diff-right">List(...)</div></summary><div class="complex-value list"><table><tr><td><span class="object-key int">0</span></td><td><details class="pyglove dict"><summary><div class="summary-title">Dict(...)</div></summary><div class="complex-value dict"><details open class="pyglove int"><summary><div class="summary-name">x</div><div class="summary-title">int</div></summary><span class="simple-value int">3</span></details></div></details></td></tr></table></div></details></div></td></tr></table></div></details>
+        """
+    )
+
 
 if __name__ == '__main__':
   unittest.main()

pyglove/core/symbolic/flags.py

@@ -14,7 +14,7 @@
 """Global, thread-local and scoped flags for handling symbolic objects."""
 
 from typing import Any, Callable, ContextManager, Optional
-from pyglove.core.object_utils import thread_local
+from pyglove.core.utils import thread_local
 
 
 #