From edd55388ce1202a4df6a0186b655787c72108fd2 Mon Sep 17 00:00:00 2001 From: dc-dc-dc Date: Wed, 20 Dec 2023 13:56:55 -0500 Subject: [PATCH] switch python load test to use load --- python/tests/test_load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tests/test_load.py b/python/tests/test_load.py index 7c05906cf..c38fff661 100644 --- a/python/tests/test_load.py +++ b/python/tests/test_load.py @@ -84,7 +84,7 @@ class TestLoad(mlx_tests.MLXTestCase): with open(save_file_mlx, "wb") as f: mx.save_safetensor(f, save_dict) with open(save_file_mlx, "rb") as f: - load_dict = mx.load_safetensor(f) + load_dict = mx.load(f) self.assertTrue("test" in load_dict) self.assertTrue(