Return reference_value_shape as default

firedrakeproject · Nov 14, 2024 · 769c918 · 769c918
1 parent 3be472e
commit 769c918
Showing 1 changed file with 3 additions and 5 deletions.
diff --git 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()