diff --git a/ufl/pullback.py b/ufl/pullback.py index c2bf18d84..0dc780e52 100644 --- a/ufl/pullback.py +++ b/ufl/pullback.py @@ -560,7 +560,7 @@ def physical_value_shape(self, element, domain) -> typing.Tuple[int, ...]: Returns: The value shape when the pull back is applied to the given element """ - raise NotImplementedError() + return element.reference_value_shape class CustomPullback(AbstractPullback): @@ -598,9 +598,7 @@ def physical_value_shape(self, element, domain) -> typing.Tuple[int, ...]: Returns: The value shape when the pull back is applied to the given element """ - if element.reference_value_shape == (): - return () - raise NotImplementedError() + return element.reference_value_shape class UndefinedPullback(AbstractPullback): @@ -628,7 +626,7 @@ def physical_value_shape(self, element, domain) -> typing.Tuple[int, ...]: Returns: The value shape when the pull back is applied to the given element """ - raise NotImplementedError() + return element.reference_value_shape identity_pullback = IdentityPullback()