Fix default variant

firedrakeproject · Oct 24, 2024 · f1a5629 · f1a5629
1 parent 6914a63
commit f1a5629
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/FIAT/hellan_herrmann_johnson.py b/FIAT/hellan_herrmann_johnson.py
@@ -97,8 +97,10 @@ class HellanHerrmannJohnson(finite_element.CiarletElement):
        HHJ(r) is the space of symmetric-matrix-valued polynomials of degree r
        or less with normal-normal continuity.
     """
-    def __init__(self, ref_el, degree, variant="integral"):
+    def __init__(self, ref_el, degree, variant=None):
         assert degree >= 0, "Hellan-Herrmann-Johnson starts at degree 0!"
+        if variant is None:
+            variant = "integral"
         poly_set = polynomial_set.ONSymTensorPolynomialSet(ref_el, degree)
         dual = HellanHerrmannJohnsonDual(ref_el, degree, variant)
         sd = ref_el.get_spatial_dimension()

diff --git a/FIAT/regge.py b/FIAT/regge.py
@@ -62,8 +62,10 @@ class Regge(finite_element.CiarletElement):
        REG(r) is the space of symmetric-matrix-valued polynomials of degree r
        or less with tangential-tangential continuity.
     """
-    def __init__(self, ref_el, degree, variant="integral"):
+    def __init__(self, ref_el, degree, variant=None):
         assert degree >= 0, "Regge start at degree 0!"
+        if variant is None:
+            variant = "integral"
         poly_set = polynomial_set.ONSymTensorPolynomialSet(ref_el, degree)
         dual = ReggeDual(ref_el, degree, variant)
         formdegree = (1, 1)