diff --git a/tests/eager_test.py b/tests/eager_test.py index a39a455f36..e8dd5c2e74 100644 --- a/tests/eager_test.py +++ b/tests/eager_test.py @@ -163,9 +163,9 @@ def test_dft_cfft_last_axis(self): np.testing.assert_allclose(expected1, expected2) with self.subTest( c_shape=c.shape, - le=list(le), + le=le.tolist(), expected_shape=expected1.shape, - weights=we, + weights=we.tolist(), ): case = onnx_script_test_case.FunctionTestParams( signal_dft.dft_last_axis, [x, le, False], [expected1] @@ -192,7 +192,7 @@ def test_dft_rfft(self, x_, s: int): nax = np.array([ax], dtype=np.int64) with self.subTest( x_shape=x.shape, - le=list(le), + le=le.tolist(), ax=ax, expected_shape=expected.shape, ): @@ -230,7 +230,7 @@ def test_dft_cfft(self, x, y): np.testing.assert_allclose(expected1, expected2) with self.subTest( c_shape=c.shape, - le=list(le), + le=le.tolist(), ax=ax, expected_shape=expected1.shape, ): @@ -256,7 +256,7 @@ def test_dft_rifft(self, x_): nax = np.array([ax], dtype=np.int64) with self.subTest( x_shape=x.shape, - le=list(le), + le=le.tolist(), ax=str(ax), expected_shape=expected.shape, ): @@ -295,7 +295,7 @@ def test_dft_cifft(self, x, y): np.testing.assert_allclose(expected1, expected2) with self.subTest( c_shape=c.shape, - le=list(le), + le=le.tolist(), ax=str(ax), expected_shape=expected1.shape, ):