Skip to content

Commit

Permalink
style
Browse files Browse the repository at this point in the history
  • Loading branch information
pbrubeck committed Oct 15, 2023
1 parent 06ff136 commit 49bee1e
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 6 deletions.
11 changes: 6 additions & 5 deletions FIAT/gauss_legendre.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,11 +49,12 @@ def __init__(self, ref_el, degree):
raise ValueError("Gauss-Legendre elements are only defined on simplices.")
dual = GaussLegendreDualSet(ref_el, degree)
if ref_el.shape == LINE:
pts = []
for phi in dual.nodes:
pt, = phi.pt_dict.keys()
pts.append(pt)
poly_set = LagrangePolynomialSet(ref_el, pts)
points = []
for node in dual.nodes:
# Assert singleton point for each node.
pt, = node.get_point_dict().keys()
points.append(pt)
poly_set = LagrangePolynomialSet(ref_el, points)
else:
poly_set = polynomial_set.ONPolynomialSet(ref_el, degree)
formdegree = ref_el.get_spatial_dimension() # n-form
Expand Down
2 changes: 1 addition & 1 deletion FIAT/gauss_lobatto_legendre.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ def __init__(self, ref_el, degree):
points = []
for node in dual.nodes:
# Assert singleton point for each node.
pt, = node.pt_dict.keys()
pt, = node.get_point_dict().keys()
points.append(pt)
poly_set = LagrangePolynomialSet(ref_el, points)
else:
Expand Down

0 comments on commit 49bee1e

Please sign in to comment.