From 204e4af631788be395dc80bba78b5759e210be64 Mon Sep 17 00:00:00 2001
From: Alexandru Fikl <alexfikl@gmail.com>
Date: Sat, 8 Feb 2025 15:21:06 +0200
Subject: [PATCH] fix: add __array_ufunc__ to containers in examples

---
 examples/parallel-vtkhdf.py | 2 +-
 examples/simple-dg.py       | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/examples/parallel-vtkhdf.py b/examples/parallel-vtkhdf.py
index 3ff980e..287597b 100644
--- a/examples/parallel-vtkhdf.py
+++ b/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"
diff --git a/examples/simple-dg.py b/examples/simple-dg.py
index 51da62f..efc396b 100644
--- a/examples/simple-dg.py
+++ b/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"
 
-- 
GitLab