revert num_exp_function to use polynomial_dimension

firedrakeproject · Nov 9, 2023 · 05d36c9 · 05d36c9
1 parent b0173d4
commit 05d36c9
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/FIAT/polynomial_set.py b/FIAT/polynomial_set.py
@@ -127,10 +127,10 @@ def __init__(self, ref_el, degree, shape=tuple()):
         else:
             flat_shape = numpy.ravel(shape)
             num_components = numpy.prod(flat_shape)
-        expansion_set = expansions.ExpansionSet(ref_el)
-        num_exp_functions = expansion_set.get_num_members(degree)
+        num_exp_functions = expansions.polynomial_dimension(ref_el, degree)
         num_members = num_components * num_exp_functions
         embedded_degree = degree
+        expansion_set = expansions.ExpansionSet(ref_el)
 
         if shape == tuple():
             coeffs = numpy.eye(num_members)
@@ -219,11 +219,11 @@ def __init__(self, ref_el, degree, size=None):
             size = sd
 
         shape = (size, size)
-        expansion_set = expansions.ExpansionSet(ref_el)
-        num_exp_functions = expansion_set.get_num_members(degree)
+        num_exp_functions = expansions.polynomial_dimension(ref_el, degree)
         num_components = size * (size + 1) // 2
         num_members = num_components * num_exp_functions
         embedded_degree = degree
+        expansion_set = expansions.ExpansionSet(ref_el)
 
         # set up coefficients for symmetric tensors
         coeffs_shape = (num_members, *shape, num_exp_functions)