@@ -163,9 +163,9 @@ def test_dft_cfft_last_axis(self):
163
163
np .testing .assert_allclose (expected1 , expected2 )
164
164
with self .subTest (
165
165
c_shape = c .shape ,
166
- le = list ( le ),
166
+ le = le . tolist ( ),
167
167
expected_shape = expected1 .shape ,
168
- weights = we ,
168
+ weights = we . tolist () ,
169
169
):
170
170
case = onnx_script_test_case .FunctionTestParams (
171
171
signal_dft .dft_last_axis , [x , le , False ], [expected1 ]
@@ -192,7 +192,7 @@ def test_dft_rfft(self, x_, s: int):
192
192
nax = np .array ([ax ], dtype = np .int64 )
193
193
with self .subTest (
194
194
x_shape = x .shape ,
195
- le = list ( le ),
195
+ le = le . tolist ( ),
196
196
ax = ax ,
197
197
expected_shape = expected .shape ,
198
198
):
@@ -230,7 +230,7 @@ def test_dft_cfft(self, x, y):
230
230
np .testing .assert_allclose (expected1 , expected2 )
231
231
with self .subTest (
232
232
c_shape = c .shape ,
233
- le = list ( le ),
233
+ le = le . tolist ( ),
234
234
ax = ax ,
235
235
expected_shape = expected1 .shape ,
236
236
):
@@ -256,7 +256,7 @@ def test_dft_rifft(self, x_):
256
256
nax = np .array ([ax ], dtype = np .int64 )
257
257
with self .subTest (
258
258
x_shape = x .shape ,
259
- le = list ( le ),
259
+ le = le . tolist ( ),
260
260
ax = str (ax ),
261
261
expected_shape = expected .shape ,
262
262
):
@@ -295,7 +295,7 @@ def test_dft_cifft(self, x, y):
295
295
np .testing .assert_allclose (expected1 , expected2 )
296
296
with self .subTest (
297
297
c_shape = c .shape ,
298
- le = list ( le ),
298
+ le = le . tolist ( ),
299
299
ax = str (ax ),
300
300
expected_shape = expected1 .shape ,
301
301
):
0 commit comments