@@ -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