From 0232db294d4c516140eb323de83dc199d2a3d9cf Mon Sep 17 00:00:00 2001 From: Christopher Mauri Milan Date: Mon, 11 Dec 2023 19:14:00 -0800 Subject: [PATCH] fix tolist issue (#2723) --- test/test_dtype.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_dtype.py b/test/test_dtype.py index 4036dbc175..c4126a66ea 100644 --- a/test/test_dtype.py +++ b/test/test_dtype.py @@ -42,7 +42,7 @@ def _assert_eq(tensor:Tensor, target_dtype:DType, target): raise AssertionError(f"\ntensor {tensor.numpy()} dtype {tensor.dtype} does not match target {target} with dtype {target_dtype}") from e def _test_op(fxn, target_dtype:DType, target): _assert_eq(fxn(), target_dtype, target) -def _test_cast(a:Tensor, target_dtype:DType): _test_op(lambda: a.cast(target_dtype), target_dtype, a.numpy().astype(target_dtype.np).tolist()) +def _test_cast(a:Tensor, target_dtype:DType): _test_op(lambda: a.cast(target_dtype), target_dtype, list(a.numpy().astype(target_dtype.np))) def _test_bitcast(a:Tensor, target_dtype:DType, target=None): _test_op(lambda: a.bitcast(target_dtype), target_dtype, target or a.numpy().view(target_dtype.np).tolist()) class TestDType(unittest.TestCase):