add variant for MixedElement

FEniCS · Nov 1, 2022 · 51af898 · 51af898
1 parent 844154e
commit 51af898
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ufl/finiteelement/mixedelement.py b/ufl/finiteelement/mixedelement.py
@@ -234,6 +234,13 @@ def degree(self, component=None):
     def reconstruct(self, **kwargs):
         return MixedElement(*[e.reconstruct(**kwargs) for e in self.sub_elements()])
 
+    def variant(self):
+        try:
+            variant, = {e.variant() for e in self.sub_elements()}
+            return variant
+        except ValueError:
+            return None
+
     def __str__(self):
         "Format as string for pretty printing."
         tmp = ", ".join(str(element) for element in self._sub_elements)