3333else :
3434 map_coordinates = Mock ()
3535
36- from typing import Dict , List , Tuple
36+ from typing import Dict , List , Tuple , Union
3737
3838__all__ = [
3939 "distortion_correction_proj_discorpy" ,
4848def distortion_correction_proj_discorpy (
4949 data : cp .ndarray ,
5050 metadata_path : str ,
51- shift : Tuple [ int , int ] = ( 0 , 0 ) ,
52- step : Tuple [ int , int ] = ( 1 , 1 ) ,
51+ shift : List [ Union [ int , int ]] = [ 0 , 0 ] ,
52+ step : List [ Union [ int , int ]] = [ 1 , 1 ] ,
5353 order : int = 1 ,
5454 mode : str = "reflect" ,
5555):
@@ -64,10 +64,10 @@ def distortion_correction_proj_discorpy(
6464 The path to the file containing the distortion coefficients for the
6565 data.
6666
67- shift: tuple , optional
67+ shift: List , optional
6868 Centers of distortion in x (from the left of the image) and y directions (from the top of the image).
6969
70- step: tuple , optional
70+ step: List , optional
7171 Steps in x and y directions respectively. They need to be not larger than one.
7272
7373 order : int, optional.
@@ -97,8 +97,8 @@ def distortion_correction_proj_discorpy(
9797 if det_y_step > 1 or det_x_step > 1 :
9898 msg = (
9999 "\n ***********************************************\n "
100- "!!! ERROR !!! -> Method doesn't work with the step in "
101- " the preview larger than 1 \n "
100+ "!!! ERROR !!! -> Method doesn't work with the step parameter "
101+ " larger than 1 \n "
102102 "***********************************************\n "
103103 )
104104 raise ValueError (msg )
0 commit comments