Merge remote-tracking branch 'upstream/main' into ksagiyam/merge_upst…

…ream
firedrakeproject · Nov 14, 2024 · 5d16f59 · 5d16f59
2 parents 2a22650 + d5c7b33
commit 5d16f59
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ufl/pullback.py b/ufl/pullback.py
@@ -519,8 +519,10 @@ def physical_value_shape(self, element, domain) -> typing.Tuple[int, ...]:
         Returns:
             The value shape when the pull back is applied to the given element
         """
-        assert element == self._element
-        return tuple(i + 1 for i in max(self._symmetry.keys()))
+        assert isinstance(element, type(self._element))
+        subelem = element.sub_elements[0]
+        pvs = subelem.pullback.physical_value_shape(subelem, domain)
+        return tuple(i + 1 for i in max(self._symmetry.keys())) + pvs
 
 
 class PhysicalPullback(AbstractPullback):