Skip to content

Commit 321cb41

Browse files
authored
[CI] Fix execnet.gateway_base.DumpError: can't serialize <class 'numpy.int64'> (#2379)
1 parent c23e512 commit 321cb41

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/eager_test.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -163,9 +163,9 @@ def test_dft_cfft_last_axis(self):
163163
np.testing.assert_allclose(expected1, expected2)
164164
with self.subTest(
165165
c_shape=c.shape,
166-
le=list(le),
166+
le=le.tolist(),
167167
expected_shape=expected1.shape,
168-
weights=we,
168+
weights=we.tolist(),
169169
):
170170
case = onnx_script_test_case.FunctionTestParams(
171171
signal_dft.dft_last_axis, [x, le, False], [expected1]
@@ -192,7 +192,7 @@ def test_dft_rfft(self, x_, s: int):
192192
nax = np.array([ax], dtype=np.int64)
193193
with self.subTest(
194194
x_shape=x.shape,
195-
le=list(le),
195+
le=le.tolist(),
196196
ax=ax,
197197
expected_shape=expected.shape,
198198
):
@@ -230,7 +230,7 @@ def test_dft_cfft(self, x, y):
230230
np.testing.assert_allclose(expected1, expected2)
231231
with self.subTest(
232232
c_shape=c.shape,
233-
le=list(le),
233+
le=le.tolist(),
234234
ax=ax,
235235
expected_shape=expected1.shape,
236236
):
@@ -256,7 +256,7 @@ def test_dft_rifft(self, x_):
256256
nax = np.array([ax], dtype=np.int64)
257257
with self.subTest(
258258
x_shape=x.shape,
259-
le=list(le),
259+
le=le.tolist(),
260260
ax=str(ax),
261261
expected_shape=expected.shape,
262262
):
@@ -295,7 +295,7 @@ def test_dft_cifft(self, x, y):
295295
np.testing.assert_allclose(expected1, expected2)
296296
with self.subTest(
297297
c_shape=c.shape,
298-
le=list(le),
298+
le=le.tolist(),
299299
ax=str(ax),
300300
expected_shape=expected1.shape,
301301
):

0 commit comments

Comments
 (0)