fix: add __array_ufunc__ to containers in examples

Author: alexfikl

examples/parallel-vtkhdf.py

@@ -73,7 +73,7 @@ def main(*, ambient_dim: int) -> None:
     logger.info("[%4d] fields: finished", comm.rank)
 
     from meshmode.discretization.visualization import make_visualizer
-    vis = make_visualizer(actx, discr, vis_order=order, force_equidistant=False)
+    vis = make_visualizer(actx, discr, force_equidistant=False)
     logger.info("[%4d] make_visualizer: finished", comm.rank)
 
     filename = f"parallel-vtkhdf-example-{ambient_dim}d.hdf"

examples/simple-dg.py

@@ -341,6 +341,9 @@ class TracePair:
     interior: ArrayContainer
     exterior: ArrayContainer
 
+    # NOTE: let the container do the broadcasting + arithmetic
+    __array_ufunc__ = None
+
     def __getattr__(self, name):
         return map_array_container(
                 lambda ary: getattr(ary, name),
@@ -450,6 +453,9 @@ class WaveState:
     u: DOFArray
     v: np.ndarray  # [object]
 
+    # NOTE: let the container do the broadcasting + arithmetic
+    __array_ufunc__ = None
+
     def __post_init__(self):
         assert isinstance(self.v, np.ndarray) and self.v.dtype.char == "O"