diff --git a/FIAT/polynomial_set.py b/FIAT/polynomial_set.py index d1ecf00c7..4242b6266 100644 --- 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,