77import numpy as np
88import numpy .typing as npt
99
10+
1011def add_float (i : float , j : float ) -> float :
1112 """Add two floats.
1213
@@ -19,7 +20,8 @@ def add_float(i: float, j: float) -> float:
1920 """
2021 return i + j
2122
22- def add_double (i : 'double' , j : 'double' ) -> 'double' :
23+
24+ def add_double (i : "double" , j : "double" ) -> "double" :
2325 """Add two doubles.
2426
2527 Args:
@@ -31,7 +33,8 @@ def add_double(i: 'double', j: 'double') -> 'double':
3133 """
3234 return i + j
3335
34- def add_unsigned (i : 'unsigned int' , j : 'unsigned int' ) -> 'unsigned int' :
36+
37+ def add_unsigned (i : "unsigned int" , j : "unsigned int" ) -> "unsigned int" :
3538 """Add two unsigned integers.
3639
3740 Args:
@@ -43,6 +46,7 @@ def add_unsigned(i: 'unsigned int', j: 'unsigned int') -> 'unsigned int':
4346 """
4447 return i + j
4548
49+
4650def collect_float (i : float , j : float ) -> npt .NDArray [np .float32 ]:
4751 """Combine floats into a numpy array.
4852
@@ -55,7 +59,8 @@ def collect_float(i: float, j: float) -> npt.NDArray[np.float32]:
5559 """
5660 return np .array ([i , j ], dtype = np .float32 )
5761
58- def collect_double (i : 'double' , j : 'double' ) -> npt .NDArray [np .float64 ]:
62+
63+ def collect_double (i : "double" , j : "double" ) -> npt .NDArray [np .float64 ]:
5964 """Combine doubles into a numpy array.
6065
6166 Args:
@@ -67,6 +72,7 @@ def collect_double(i: 'double', j: 'double') -> npt.NDArray[np.float64]:
6772 """
6873 return np .array ([i , j ], dtype = np .float64 )
6974
75+
7076def PHLEX_EXPERIMENTAL_REGISTER_ALGORITHMS (m , config ):
7177 """Register algorithms.
7278
@@ -77,22 +83,24 @@ def PHLEX_EXPERIMENTAL_REGISTER_ALGORITHMS(m, config):
7783 Returns:
7884 None
7985 """
80- m .transform (add_float ,
81- input_family = config ["input_float" ],
82- output_products = config ["output_float" ])
83-
84- m .transform (add_double ,
85- input_family = config ["input_double" ],
86- output_products = config ["output_double" ])
87-
88- m .transform (add_unsigned ,
89- input_family = config ["input_uint" ],
90- output_products = config ["output_uint" ])
91-
92- m .transform (collect_float ,
93- input_family = config ["input_float" ],
94- output_products = config ["output_vfloat" ])
95-
96- m .transform (collect_double ,
97- input_family = config ["input_double" ],
98- output_products = config ["output_vdouble" ])
86+ m .transform (
87+ add_float , input_family = config ["input_float" ], output_products = config ["output_float" ]
88+ )
89+
90+ m .transform (
91+ add_double , input_family = config ["input_double" ], output_products = config ["output_double" ]
92+ )
93+
94+ m .transform (
95+ add_unsigned , input_family = config ["input_uint" ], output_products = config ["output_uint" ]
96+ )
97+
98+ m .transform (
99+ collect_float , input_family = config ["input_float" ], output_products = config ["output_vfloat" ]
100+ )
101+
102+ m .transform (
103+ collect_double ,
104+ input_family = config ["input_double" ],
105+ output_products = config ["output_vdouble" ],
106+ )
0 commit comments