Skip to content

Commit 413298e

Browse files
committed
fix format for ruff precomit hook
1 parent 2a77292 commit 413298e

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

src/numbacs/integration.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -121,9 +121,7 @@ def flowmap_n(funcptr, t0, T, pts, params, method="dop853", n=2, rtol=1e-6, atol
121121

122122

123123
@njit(parallel=True)
124-
def flowmap_grid_2D(
125-
funcptr, t0, T, x, y, params, method="dop853", rtol=1e-6, atol=1e-8, mask=None
126-
):
124+
def flowmap_grid_2D(funcptr, t0, T, x, y, params, method="dop853", rtol=1e-6, atol=1e-8, mask=None):
127125
"""
128126
Computes the flow map at the final time of the ode defined by funcptr where funcptr is a
129127
pointer to a C callback created within numba using the ``@cfunc`` decorator. Flow map is
@@ -636,12 +634,12 @@ def flowmap_composition(flowmaps, grid, nT):
636634
fy = eval_linear(grid, flowmaps[k, :, :, 1], pts, xto.CONSTANT)
637635
pts = np.column_stack((fx, fy))
638636

639-
composed_flowmap[:, :, 0] = eval_linear(
640-
grid, flowmaps[-1, :, :, 0], pts, xto.CONSTANT
641-
).reshape(nx, ny)
642-
composed_flowmap[:, :, 1] = eval_linear(
643-
grid, flowmaps[-1, :, :, 1], pts, xto.CONSTANT
644-
).reshape(nx, ny)
637+
composed_flowmap[:, :, 0] = eval_linear(grid, flowmaps[-1, :, :, 0], pts, xto.CONSTANT).reshape(
638+
nx, ny
639+
)
640+
composed_flowmap[:, :, 1] = eval_linear(grid, flowmaps[-1, :, :, 1], pts, xto.CONSTANT).reshape(
641+
nx, ny
642+
)
645643

646644
return composed_flowmap
647645

0 commit comments

Comments
 (0)