zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-21 20:46:46 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

switch python load test to use load

Browse Source
This commit is contained in:
dc-dc-dc 2023-12-20 13:56:55 -05:00
parent dc5abdc4c4
commit edd55388ce
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_load.py
View File

@ -84,7 +84,7 @@ class TestLoad(mlx_tests.MLXTestCase):
with open(save_file_mlx, "wb") as f: with open(save_file_mlx, "wb") as f:
mx.save_safetensor(f, save_dict) mx.save_safetensor(f, save_dict)
with open(save_file_mlx, "rb") as f: 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("test" in load_dict)
self.assertTrue( self.assertTrue(