diff --git a/pytential/symbolic/execution.py b/pytential/symbolic/execution.py index a0e6e9f17..378951cbe 100644 --- a/pytential/symbolic/execution.py +++ b/pytential/symbolic/execution.py @@ -300,10 +300,10 @@ def map_is_shape_class(self, expr): raise NotImplementedError( "non-homogeneous element groups are not supported") - from meshmode.mesh import _ModepyElementGroup + from meshmode.mesh import ModepyElementGroup meg = discr.groups[0].mesh_el_group - if isinstance(meg, _ModepyElementGroup): - return isinstance(meg._modepy_shape, expr.shape) + if isinstance(meg, ModepyElementGroup): + return isinstance(meg.shape, expr.shape) else: raise TypeError(f"element type not supported: '{type(meg).__name__}'")