@@ -347,24 +347,24 @@ def grid_boundary_search(
347347 if offset is None :
348348 offset = [None for _ in np .arange (len (obspt ))]
349349
350- if "floodfill" in kwargs and kwargs [ "floodfill" ] :
350+ if kwargs . get ( "floodfill" ) :
351351 s_ffallrefpts = []
352352 g_ffallrefpts = []
353- for obs , orbit in zip (obspt , offset ):
353+ for obs , orbit in zip (obspt , offset , strict = True ):
354354 # set ring and offsets
355- obs = 0 if obs is None else obs
356- dpp = 0.0 if dp is None else dp
357- orbit , ringrot = set_ring_orbit (ring , dpp , obs , orbit )
358- offset = orbit + np .array ([0 , 0 , 0 , 0 , dpp , 0 ])
355+ _obs = 0 if obs is None else obs
356+ _dpp = 0.0 if dp is None else dp
357+ _orbit , _ringrot = set_ring_orbit (ring , _dpp , _obs , orbit )
358+ _offset = _orbit + np .array ([0 , 0 , 0 , 0 , _dpp , 0 ])
359359 # flood fill
360360 window = np .ravel ((np .array (config .bounds ).T * config .amplitudes ).T )
361361 data_ff = floodfill (
362- ringrot ,
362+ _ringrot ,
363363 nturns = nturns ,
364364 window = window ,
365365 grid_size = config .shape ,
366366 axes = config .planesi ,
367- offset = offset ,
367+ offset = _offset ,
368368 verbose = False ,
369369 use_mp = use_mp ,
370370 pool_size = kwargs ["pool_size" ],
0 commit comments