From 5c24928b7fc6d6561f574a63c0a47e771f19a895 Mon Sep 17 00:00:00 2001 From: Pablo Brubeck Date: Sat, 14 Dec 2024 17:02:17 +0000 Subject: [PATCH] imex: style --- irksome/imex.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/irksome/imex.py b/irksome/imex.py index 6d71c56..7126f99 100644 --- a/irksome/imex.py +++ b/irksome/imex.py @@ -15,19 +15,18 @@ def riia_explicit_coeffs(k): of a RadauIIA-IMEX method.""" U = FIAT.ufc_simplex(1) L = FIAT.GaussRadau(U, k - 1) - Q = FIAT.make_quadrature(L.ref_el, 2*k) - c = np.asarray([list(ell.pt_dict.keys())[0][0] - for ell in L.dual.nodes]) Q = FIAT.make_quadrature(L.ref_el, 2*k) qpts = Q.get_points() qwts = Q.get_weights() A = np.zeros((k, k)) - for i in range(k): - qpts_i = 1 + qpts * c[i] - qwts_i = qwts * c[i] - Lvals_i = L.tabulate(0, qpts_i)[0, ] + for i, ell in enumerate(L.dual.nodes): + pt, = ell.pt_dict + ci, = pt + qpts_i = 1 + qpts * ci + qwts_i = qwts * ci + Lvals_i = L.tabulate(0, qpts_i)[(0,)] A[i, :] = Lvals_i @ qwts_i return A