Add set of continuous families not requiring a restriction (#73)

FEniCS · Nov 22, 2021 · 22c21bd · 22c21bd
1 parent a4b7ab0
commit 22c21bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ufl/algorithms/apply_restrictions.py b/ufl/algorithms/apply_restrictions.py
@@ -132,7 +132,8 @@ def coefficient(self, o):
         d = e.degree()
         f = e.family()
         # TODO: Move this choice to the element class?
-        if (f == "Lagrange" and d > 0) or f == "Real":
+        continuous_families = ["Lagrange", "Q", "S"]
+        if (f in continuous_families and d > 0) or f == "Real":
             # If the coefficient _value_ is _fully_ continuous
             return self._default_restricted(o)  # Must still be computed from one of the sides, we just don't care which
         else: