Unlazy dmats for NodalEnrichedElement

firedrakeproject · Oct 29, 2023 · cc48c59 · cc48c59
1 parent 28d35bc
commit cc48c59
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/FIAT/polynomial_set.py b/FIAT/polynomial_set.py
@@ -54,14 +54,14 @@ class PolynomialSet(object):
          function.
     """
 
-    def __init__(self, ref_el, degree, embedded_degree, expansion_set, coeffs):
+    def __init__(self, ref_el, degree, embedded_degree, expansion_set, coeffs, dmats=None):
         self.ref_el = ref_el
         self.num_members = coeffs.shape[0]
         self.degree = degree
         self.embedded_degree = embedded_degree
         self.expansion_set = expansion_set
         self.coeffs = coeffs
-        self.dmats = []
+        self.dmats = dmats or []
 
     def tabulate_new(self, pts):
         return numpy.dot(self.coeffs,