CR: allow variants

firedrakeproject · Jan 13, 2025 · 11f90c2 · 11f90c2
1 parent 14a66ab
commit 11f90c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/FIAT/crouzeix_raviart.py b/FIAT/crouzeix_raviart.py
@@ -63,7 +63,6 @@ def __init__(self, ref_el, degree, variant, interpolant_deg):
                         pts = make_lattice(verts, degree-1, variant="gl")
                     else:
                         pts = ref_el.make_points(dim, i, degree, variant="gll")
-
                     nodes.extend(functional.PointEvaluation(ref_el, x) for x in pts)
                     entity_ids[dim][i].extend(range(cur, len(nodes)))
 

diff --git a/finat/fiat_elements.py b/finat/fiat_elements.py
@@ -356,8 +356,8 @@ def __init__(self, cell, degree):
 
 
 class CrouzeixRaviart(ScalarFiatElement):
-    def __init__(self, cell, degree):
-        super().__init__(FIAT.CrouzeixRaviart(cell, degree))
+    def __init__(self, cell, degree, variant=None):
+        super().__init__(FIAT.CrouzeixRaviart(cell, degree, variant=variant))
 
 
 class Lagrange(ScalarFiatElement):