From bc2a29f03355b137b6197e0810557522f7bb3cd2 Mon Sep 17 00:00:00 2001
From: Awni Hannun <awni@apple.com>
Date: Fri, 6 Dec 2024 10:48:58 -0800
Subject: [PATCH] fix (#1654)

---
 .pre-commit-config.yaml    | 3 ---
 python/tests/test_array.py | 2 ++
 2 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index bd3d21d707..0f03d36d94 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -1,6 +1,3 @@
-default_language_version:
-  python: python3.10
-
 repos:
 -   repo: https://github.com/pre-commit/mirrors-clang-format
     rev: v19.1.4
diff --git a/python/tests/test_array.py b/python/tests/test_array.py
index b0c915de1a..1e1ef8bdb7 100644
--- a/python/tests/test_array.py
+++ b/python/tests/test_array.py
@@ -1,5 +1,6 @@
 # Copyright © 2023-2024 Apple Inc.
 
+import gc
 import operator
 import pickle
 import resource
@@ -1942,6 +1943,7 @@ class TestArray(mlx_tests.MLXTestCase):
             return b
 
         t()
+        gc.collect()
         expected = get_mem()
         for _ in range(100):
             t()