pyglove 0.4.5.dev202410100808__py3-none-any.whl → 0.4.5.dev202410110808__py3-none-any.whl

pyglove/core/symbolic/diff.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 """Symbolic differences."""
 
-from typing import Any, Callable, Optional, Sequence, Tuple, Union
+from typing import Any, Callable, List, Optional, Tuple, Union
 
 from pyglove.core import object_utils
 from pyglove.core import typing as pg_typing
@@ -196,25 +196,26 @@ class Diff(PureSymbolic, pg_object.Object):
       view: html.HtmlTreeView,
       parent: Any,
       root_path: object_utils.KeyPath,
-      css_class: Optional[Sequence[str]] = None,
       **kwargs
   ) -> html.Html:
-    del parent
-    user_specified_css_class = css_class or []
     if not bool(self):
       if self.value == Diff.MISSING:
         root = html.Html.element(
             'span',
             # CSS class already defined in HtmlTreeView.
-            css_class=['diff_empty'] + user_specified_css_class,
+            css_class=['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,
-            css_class=['diff_left', 'diff_right'] + user_specified_css_class,
-            root_path=root_path, **kwargs
+        root = html.Html.element(
+            'div',
+            [
+                view.content(
+                    self.value, parent=parent, root_path=root_path, **kwargs,
+                )
+            ],
+            css_class=['diff_left', 'diff_right']
         )
     elif self.is_leaf:
       root = html.Html.element(
@@ -241,7 +242,7 @@ class Diff(PureSymbolic, pg_object.Object):
                   css_class=['diff_right'],
               )  if self.right != Diff.MISSING else None,
           ],
-          css_class=['diff_value'] + user_specified_css_class,
+          css_class=['diff_value']
       )
     else:
       assert isinstance(self.left, type)
@@ -279,10 +280,14 @@ class Diff(PureSymbolic, pg_object.Object):
           ],
           css_class=[
               'complex_value', view.css_class_name(self.left)
-          ] + user_specified_css_class,
+          ]
       )
-    return root.add_style(
+    return root
+
+  def _html_style(self) -> List[str]:
+    return [
         """
+        /* Diff styles. */
         .diff .summary_title::after {
           content: ' (diff)';
           color: #aaa;
@@ -331,7 +336,7 @@ class Diff(PureSymbolic, pg_object.Object):
           background-color: #ccffcc;
         }
         """
-    )
+    ]
 
 
 # NOTE(daiyip): we add the symbolic attribute to Diff after its declaration

pyglove/core/symbolic/diff_test.py

@@ -403,6 +403,26 @@ class DiffTest(unittest.TestCase):
           color: darkred;
           font-style: italic;
         }
+        /* Value details styles. */
+        details.pyglove {
+          border: 1px solid #aaa;
+          border-radius: 4px;
+          padding: 0.5em 0.5em 0;
+          margin: 0.1em 0;
+        }
+        details.pyglove.special_value {
+          margin-bottom: 0.75em;
+        }
+        details.pyglove[open] {
+          padding: 0.5em 0.5em 0.5em;
+        }
+        .highlight {
+          background-color: Mark;
+        }
+        .lowlight {
+          opacity: 0.2;
+        }
+        /* Diff styles. */
         .diff .summary_title::after {
           content: ' (diff)';
           color: #aaa;
@@ -450,25 +470,6 @@ class DiffTest(unittest.TestCase):
         .diff_value .diff_right > details {
           background-color: #ccffcc;
         }
-        /* Value details styles. */
-        details.pyglove {
-          border: 1px solid #aaa;
-          border-radius: 4px;
-          padding: 0.5em 0.5em 0;
-          margin: 0.1em 0;
-        }
-        details.pyglove.special_value {
-          margin-bottom: 0.75em;
-        }
-        details.pyglove[open] {
-          padding: 0.5em 0.5em 0.5em;
-        }
-        .highlight {
-          background-color: Mark;
-        }
-        .lowlight {
-          opacity: 0.2;
-        }
         </style>
         """
     )
@@ -477,7 +478,7 @@ class DiffTest(unittest.TestCase):
     assert_content(
         pg_diff(1, 1).to_html(),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">Diff</div><span class="tooltip">No diff</span></summary><span class="diff_empty"></span></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">Diff</div><span class="tooltip diff">No diff</span></summary><span class="diff_empty"></span></details>
         """
     )
 
@@ -485,7 +486,7 @@ class DiffTest(unittest.TestCase):
     assert_content(
         pg_diff(1, 1, mode='both').to_html(),
         """
-        <span class="simple_value int diff_left diff_right">1</span>
+        <div class="diff_left diff_right"><span class="simple_value int">1</span></div>
         """
     )
     # No diff complex value (diff only)
@@ -511,7 +512,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo diff_left diff_right"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">y</span></td><td><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key">0</span></td><td><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">2</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">3</span></td></tr></table></div></details></td></tr></table></div></details></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">Foo(...)</div></summary><div class="diff_left diff_right"><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key int">0</span></td><td><div><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">3</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div></td></tr></table></div></div></details>
         """
     )
 
@@ -530,7 +531,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">List</div></summary><div class="complex_value list"><table><tr><td><span class="object_key no_diff_key">0</span><span class="tooltip">[0]</span></td><td><span class="simple_value int diff_left diff_right">0</span></td></tr><tr><td><span class="object_key">1</span><span class="tooltip">[1]</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">1</span></div><div class="diff_right"><span class="simple_value int">2</span></div></div></td></tr><tr><td><span class="object_key">2</span><span class="tooltip">[2]</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div></div></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">List</div></summary><div class="complex_value list"><table><tr><td><span class="object_key int no_diff_key">0</span><span class="tooltip key-path">[0]</span></td><td><div class="diff_left diff_right"><span class="simple_value int">0</span></div></td></tr><tr><td><span class="object_key int">1</span><span class="tooltip key-path">[1]</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">1</span></div><div class="diff_right"><span class="simple_value int">2</span></div></div></td></tr><tr><td><span class="object_key int">2</span><span class="tooltip key-path">[2]</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div></div></td></tr></table></div></details>
         """
     )
 
@@ -541,7 +542,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">dict</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key no_diff_key">x</span><span class="tooltip">x</span></td><td><span class="simple_value int diff_left diff_right">1</span></td></tr><tr><td><span class="object_key">y</span><span class="tooltip">y</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div><div class="diff_right"><span class="simple_value int">3</span></div></div></td></tr><tr><td><span class="object_key">z</span><span class="tooltip">z</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">3</span></div></div></td></tr><tr><td><span class="object_key">w</span><span class="tooltip">w</span></td><td><div class="diff_value"><div class="diff_right"><span class="simple_value int">4</span></div></div></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">dict</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key str no_diff_key">x</span><span class="tooltip key-path">x</span></td><td><div class="diff_left diff_right"><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span><span class="tooltip key-path">y</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div><div class="diff_right"><span class="simple_value int">3</span></div></div></td></tr><tr><td><span class="object_key str">z</span><span class="tooltip key-path">z</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">3</span></div></div></td></tr><tr><td><span class="object_key str">w</span><span class="tooltip key-path">w</span></td><td><div class="diff_value"><div class="diff_right"><span class="simple_value int">4</span></div></div></td></tr></table></div></details>
         """
     )
 
@@ -556,7 +557,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">Foo</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span><span class="tooltip">x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div><div class="diff_right"><span class="simple_value int">1</span></div></div></td></tr><tr><td><span class="object_key">y</span><span class="tooltip">y</span></td><td><details class="pyglove diff"><summary><div class="summary_title">Foo</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span><span class="tooltip">y.x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">3</span></div><div class="diff_right"><span class="simple_value int">2</span></div></div></td></tr><tr><td><span class="object_key no_diff_key">y</span><span class="tooltip">y.y</span></td><td><span class="simple_value int diff_left diff_right">3</span></td></tr></table></div></details></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">Foo</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span><span class="tooltip key-path">x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div><div class="diff_right"><span class="simple_value int">1</span></div></div></td></tr><tr><td><span class="object_key str">y</span><span class="tooltip key-path">y</span></td><td><details class="pyglove diff"><summary><div class="summary_title">Foo</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span><span class="tooltip key-path">y.x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">3</span></div><div class="diff_right"><span class="simple_value int">2</span></div></div></td></tr><tr><td><span class="object_key str no_diff_key">y</span><span class="tooltip key-path">y.y</span></td><td><div class="diff_left diff_right"><span class="simple_value int">3</span></div></td></tr></table></div></details></td></tr></table></div></details>
         """
     )
 
@@ -571,7 +572,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <div class="diff_value"><div class="diff_left"><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">2</span></td></tr><tr><td><span class="object_key">y</span></td><td><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">3</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">3</span></td></tr></table></div></details></td></tr></table></div></details></div><div class="diff_right"><details class="pyglove bar"><summary><div class="summary_title">Bar(...)</div></summary><div class="complex_value bar"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">2</span></td></tr><tr><td><span class="object_key">y</span></td><td><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">3</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">3</span></td></tr></table></div></details></td></tr></table></div></details></div></div>
+        <div class="diff_value"><div class="diff_left"><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">3</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">3</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div><div class="diff_right"><details class="pyglove bar"><summary><div class="summary_title">Bar(...)</div></summary><div class="complex_value bar"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">3</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">3</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div></div>
         """
     )
 
@@ -586,7 +587,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key no_diff_key">x</span><span class="tooltip">x</span></td><td><span class="simple_value int diff_left diff_right">2</span></td></tr><tr><td><span class="object_key">y</span><span class="tooltip">y</span></td><td><details class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key no_diff_key">x</span><span class="tooltip">y.x</span></td><td><span class="simple_value int diff_left diff_right">3</span></td></tr><tr><td><span class="object_key no_diff_key">y</span><span class="tooltip">y.y</span></td><td><span class="simple_value int diff_left diff_right">3</span></td></tr></table></div></details></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str no_diff_key">x</span><span class="tooltip key-path">x</span></td><td><div class="diff_left diff_right"><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span><span class="tooltip key-path">y</span></td><td><details class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str no_diff_key">x</span><span class="tooltip key-path">y.x</span></td><td><div class="diff_left diff_right"><span class="simple_value int">3</span></div></td></tr><tr><td><span class="object_key str no_diff_key">y</span><span class="tooltip key-path">y.y</span></td><td><div class="diff_left diff_right"><span class="simple_value int">3</span></div></td></tr></table></div></details></td></tr></table></div></details>
         """
     )
 
@@ -601,7 +602,7 @@ class DiffTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span><span class="tooltip">x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div><div class="diff_right"><span class="simple_value int">3</span></div></div></td></tr><tr><td><span class="object_key">y</span><span class="tooltip">y</span></td><td><details class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span><span class="tooltip">y.x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">3</span></div><div class="diff_right"><span class="simple_value int">2</span></div></div></td></tr><tr><td><span class="object_key no_diff_key">y</span><span class="tooltip">y.y</span></td><td><span class="simple_value int diff_left diff_right">3</span></td></tr></table></div></details></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span><span class="tooltip key-path">x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">2</span></div><div class="diff_right"><span class="simple_value int">3</span></div></div></td></tr><tr><td><span class="object_key str">y</span><span class="tooltip key-path">y</span></td><td><details class="pyglove diff"><summary><div class="summary_title">Foo | Bar</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span><span class="tooltip key-path">y.x</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value int">3</span></div><div class="diff_right"><span class="simple_value int">2</span></div></div></td></tr><tr><td><span class="object_key str no_diff_key">y</span><span class="tooltip key-path">y.y</span></td><td><div class="diff_left diff_right"><span class="simple_value int">3</span></div></td></tr></table></div></details></td></tr></table></div></details>
         """
     )
 
@@ -623,7 +624,7 @@ class DiffTest(unittest.TestCase):
             uncollapse=['[0]', '[1].right', '[3].left.y'],
         ),
         """
-        <details open class="pyglove diff"><summary><div class="summary_title">List</div></summary><div class="complex_value list"><table><tr><td><span class="object_key no_diff_key">0</span><span class="tooltip">[0]</span></td><td><details open class="pyglove diff"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo diff_left diff_right"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">2</span></td></tr></table></div></details></td></tr><tr><td><span class="object_key">1</span><span class="tooltip">[1]</span></td><td><div class="diff_value"><div class="diff_left"><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">2</span></td></tr></table></div></details></div><div class="diff_right"><details open class="pyglove bar"><summary><div class="summary_title">Bar(...)</div></summary><div class="complex_value bar"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">2</span></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key">2</span><span class="tooltip">[2]</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value none-type">None</span></div><div class="diff_right"><details class="pyglove dict"><summary><div class="summary_title">Dict(...)</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key">3</span><span class="tooltip">[3]</span></td><td><div class="diff_value"><div class="diff_left"><details open class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">y</span></td><td><details open class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">2</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">3</span></td></tr></table></div></details></td></tr></table></div></details></div><div class="diff_right"><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key">0</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">1</span></td><td><span class="simple_value int">2</span></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key">4</span><span class="tooltip">[4]</span></td><td><div class="diff_value"><div class="diff_left"><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key">0</span></td><td><details class="pyglove dict"><summary><div class="summary_title">Dict(...)</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">2</span></td></tr></table></div></details></td></tr></table></div></details></div><div class="diff_right"><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">2</span></td></tr><tr><td><span class="object_key">y</span></td><td><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">2</span></td></tr><tr><td><span class="object_key">y</span></td><td><span class="simple_value int">4</span></td></tr></table></div></details></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key">5</span><span class="tooltip">[5]</span></td><td><div class="diff_value"><div class="diff_right"><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key">0</span></td><td><details class="pyglove dict"><summary><div class="summary_title">Dict(...)</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">3</span></td></tr></table></div></details></td></tr></table></div></details></div></div></td></tr></table></div></details>
+        <details open class="pyglove diff"><summary><div class="summary_title">List</div></summary><div class="complex_value list"><table><tr><td><span class="object_key int no_diff_key">0</span><span class="tooltip key-path">[0]</span></td><td><details open class="pyglove diff"><summary><div class="summary_title">Foo(...)</div></summary><div class="diff_left diff_right"><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">2</span></div></td></tr></table></div></div></details></td></tr><tr><td><span class="object_key int">1</span><span class="tooltip key-path">[1]</span></td><td><div class="diff_value"><div class="diff_left"><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">2</span></div></td></tr></table></div></details></div><div class="diff_right"><details open class="pyglove bar"><summary><div class="summary_title">Bar(...)</div></summary><div class="complex_value bar"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">2</span></div></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key int">2</span><span class="tooltip key-path">[2]</span></td><td><div class="diff_value"><div class="diff_left"><span class="simple_value none-type">None</span></div><div class="diff_right"><details class="pyglove dict"><summary><div class="summary_title">Dict(...)</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key int">3</span><span class="tooltip key-path">[3]</span></td><td><div class="diff_value"><div class="diff_left"><details open class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><details open class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">3</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div><div class="diff_right"><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key int">0</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key int">1</span></td><td><div><span class="simple_value int">2</span></div></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key int">4</span><span class="tooltip key-path">[4]</span></td><td><div class="diff_value"><div class="diff_left"><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key int">0</span></td><td><div><details class="pyglove dict"><summary><div class="summary_title">Dict(...)</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">2</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div><div class="diff_right"><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><details class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">2</span></div></td></tr><tr><td><span class="object_key str">y</span></td><td><div><span class="simple_value int">4</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div></div></td></tr><tr><td><span class="object_key int">5</span><span class="tooltip key-path">[5]</span></td><td><div class="diff_value"><div class="diff_right"><details class="pyglove list"><summary><div class="summary_title">List(...)</div></summary><div class="complex_value list"><table><tr><td><span class="object_key int">0</span></td><td><div><details class="pyglove dict"><summary><div class="summary_title">Dict(...)</div></summary><div class="complex_value dict"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">3</span></div></td></tr></table></div></details></div></td></tr></table></div></details></div></div></td></tr></table></div></details>
         """
     )
 

pyglove/core/symbolic/ref_test.py

@@ -223,26 +223,26 @@ class RefTest(unittest.TestCase):
           visibility: visible;
           background-color: darkblue;
         }
-        .complex_value .object_key{
+        .object_key.str {
           color: gray;
           border: 1px solid lightgray;
           background-color: ButtonFace;
           border-radius: 0.2em;
           padding: 0.3em;
         }
-        .complex_value.list .object_key{
+        .object_key.int::before{
+          content: '[';
+        }
+        .object_key.int::after{
+          content: ']';
+        }
+        .object_key.int{
           border: 0;
           color: lightgray;
           background-color: transparent;
           border-radius: 0;
           padding: 0;
         }
-        .complex_value.list .object_key::before{
-          content: '[';
-        }
-        .complex_value.list .object_key::after{
-          content: ']';
-        }
         /* Simple value styles. */
         .simple_value {
           color: blue;
@@ -294,7 +294,7 @@ class RefTest(unittest.TestCase):
             enable_key_tooltip=False,
         ),
         """
-        <details open class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><details class="pyglove ref"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key">x</span></td><td><span class="simple_value int">1</span></td></tr></table></div></details></td></tr></table></div></details>
+        <details open class="pyglove foo"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><details class="pyglove ref"><summary><div class="summary_title">Foo(...)</div></summary><div class="complex_value foo"><table><tr><td><span class="object_key str">x</span></td><td><div><span class="simple_value int">1</span></div></td></tr></table></div></details></div></td></tr></table></div></details>
         """
     )
 

pyglove/core/views/base.py

@@ -131,6 +131,7 @@ import copy as copy_lib
 import functools
 import inspect
 import io
+import os
 import types
 from typing import Any, Callable, ContextManager, Dict, Iterator, Optional, Sequence, Set, Type, Union
 
@@ -330,6 +331,7 @@ class Content(object_utils.Formattable, metaclass=abc.ABCMeta):
 
   def save(self, file: str, **kwargs):
     """Save content to a file."""
+    pg_io.mkdirs(os.path.dirname(file), exist_ok=True)
     pg_io.writefile(file, self.to_str(**kwargs))
 
   def __add__(self, other: WritableTypes) -> 'Content':

pyglove/core/views/base_test.py

@@ -209,7 +209,7 @@ class ContentTest(unittest.TestCase):
     self.assertEqual(document.content, 'abcdefghijklmno')
 
   def test_save(self):
-    filename = os.path.join(tempfile.gettempdir(), 'test_doc.txt')
+    filename = os.path.join(tempfile.gettempdir(), '1', 'test_doc.txt')
     Document('abc', ref_links=['https://x/y.css']).save(filename)
     self.assertTrue(pg_io.path_exists(filename))
     self.assertEqual(pg_io.readfile(filename), 'link=https://x/y.css\nabc')

pyglove/core/views/html/base.py

@@ -18,12 +18,17 @@ import functools
 import html as html_lib
 import inspect
 import typing
-from typing import Any, Callable, Dict, Iterable, List, Optional, Union
+from typing import Any, Callable, Dict, Iterable, List, Optional, Sequence, Union
 
 from pyglove.core import object_utils
 from pyglove.core import typing as pg_typing
 from pyglove.core.views import base
 
+NestableStr = Union[
+    str,
+    Sequence[Union[str, None, Sequence[Optional[str]]]],
+    None,
+]
 
 NodeFilter = base.NodeFilter
 NodeColor = Callable[
@@ -280,7 +285,7 @@ class Html(base.Content):
       inner_html: Optional[List[WritableTypes]] = None,
       *,
       options: Union[str, Iterable[str], None] = None,
-      css_class: Union[str, Iterable[str], None] = None,
+      css_class: NestableStr = None,
       style: Union[str, Dict[str, Any], None] = None,
       **properties
   ) -> 'Html':
@@ -301,21 +306,12 @@ class Html(base.Content):
     Returns:
       The opening tag of an HTML element.
     """
-    def ws_join(items: Union[str, Iterable[str], None]) -> Optional[str]:
-      if isinstance(items, str):
-        return items
-      elif isinstance(items, list):
-        return ' '.join(s for s in items if s is not None)
-      else:
-        assert items is None, items
-        return None
-
     s = cls()
 
     # Write the open tag.
-    css_class = ws_join(css_class)
+    css_class = cls.concate(css_class)
+    options = cls.concate(options)
     style = cls.style_str(style)
-    options = ws_join(options)
     s.write(
         f'<{tag}',
         f' {options}' if options else None,
@@ -353,9 +349,24 @@ class Html(base.Content):
           s, shared_parts_only=True
       )
 
+  @classmethod
+  def concate(
+      cls, nestable_str: NestableStr, separator: str = ' '
+  ) -> Optional[str]:
+    """Concates the string nodes in a nestable object."""
+    if isinstance(nestable_str, str):
+      return nestable_str
+    elif isinstance(nestable_str, list):
+      flattened = [cls.concate(s) for s in nestable_str]
+      flattened = [s for s in flattened if s is not None]
+      return separator.join(flattened) if flattened else None
+    else:
+      assert nestable_str is None, nestable_str
+      return None
+
   @classmethod
   def style_str(
-      cls, style: Union[str, Dict[str, Any], None] = None,
+      cls, style: Union[str, Dict[str, Any], None],
   ) -> Optional[str]:
     """Gets a string representing an inline CSS style.
 
@@ -392,6 +403,24 @@ class HtmlView(base.View):
   class Extension(base.View.Extension):
     """Base class for HtmlView extensions."""
 
+    def _html_style(self) -> List[str]:
+      """Returns additional CSS style to add for this extension.
+
+      Subclasses can override this method to add additional CSS style to the
+      rendered HTML.
+      """
+      return []
+
+    def _html_element_class(self) -> List[str]:
+      """Returns the CSS classes for the rendered element for this node.
+
+      Subclasses can override this method to add CSS classes to the
+      rendered element of this object.
+      """
+      return [
+          object_utils.camel_to_snake(self.__class__.__name__, '-')
+      ]
+
     def to_html(
         self,
         *,

pyglove/core/views/html/base_test.py

@@ -696,6 +696,15 @@ class HtmlTest(TestCase):
     self.assertEqual(Html.escape(Html('foo"bar')), Html('foo"bar'))
     self.assertEqual(Html.escape(lambda: 'foo"bar'), 'foo"bar')
 
+  def test_concate(self):
+    self.assertIsNone(Html.concate(None))
+    self.assertIsNone(Html.concate([None, [None, [None, None]]]))
+    self.assertEqual(Html.concate('a'), 'a')
+    self.assertEqual(Html.concate(['a']), 'a')
+    self.assertEqual(Html.concate(['a', None, 'b']), 'a b')
+    self.assertEqual(
+        Html.concate(['a', 'b', [None, 'c', [None, 'd']]]), 'a b c d')
+
   def test_element(self):
     # Empty element.
     self.assertEqual(Html.element('div').content, '<div></div>')