From 8dcdd1bb8ce6c9301340c9f85e5f1a63874baea0 Mon Sep 17 00:00:00 2001
From: Luca Arnaboldi <luca@arnaboldi.lu>
Date: Tue, 20 Feb 2024 17:33:29 +0100
Subject: [PATCH] Fixing typos

---
 python/tests/test_array.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/python/tests/test_array.py b/python/tests/test_array.py
index d30fd0de0..10af1ccb8 100644
--- a/python/tests/test_array.py
+++ b/python/tests/test_array.py
@@ -667,9 +667,9 @@ class TestArray(mlx_tests.MLXTestCase):
             mx.uint16,
             mx.uint32,
             mx.uint64,
-            # mx.float16,
+            mx.float16,
             mx.float32,
-            # mx.bfloat16,
+            mx.bfloat16,
             mx.complex64,
         ]
         import pickle
@@ -691,9 +691,9 @@ class TestArray(mlx_tests.MLXTestCase):
             mx.uint16,
             mx.uint32,
             mx.uint64,
-            # mx.float16,
+            mx.float16,
             mx.float32,
-            # mx.bfloat16,
+            mx.bfloat16,
             mx.complex64,
         ]
 
@@ -706,7 +706,6 @@ class TestArray(mlx_tests.MLXTestCase):
                 self.assertEqualArray(x, y)
 
                 y -= 1
-                print(x, y)
                 self.assertEqualArray(x - 1, y)
 
     def test_indexing(self):