Pickle: workaround for bfloat16

2025-10-07 16:18:09 +08:00 · 2024-03-04 23:45:17 +01:00
parent 6b6b4f0a5f
commit 25afe6bc09
2 changed files with 19 additions and 7 deletions
--- a/python/tests/test_array.py
+++ b/python/tests/test_array.py
@@ -671,7 +671,7 @@ class TestArray(mlx_tests.MLXTestCase):
            mx.uint64,
            mx.float16,
            mx.float32,
-            # mx.bfloat16,
+            mx.bfloat16,
            mx.complex64,
        ]