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

Export with callback (#2612)

Browse Source 
* export with callback

* export with callback

* Add types, fix kwarg ordering bug + test

* cleanup, test, fix

* typos
This commit is contained in:
Awni Hannun
2025-10-08 19:24:33 -07:00
committed by GitHub
parent 85a8824a8c
commit e89e8b4272
10 changed files with 370 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/tests/test_compile.py
View File

@@ -319,7 +319,7 @@ class TestCompile(mlx_tests.MLXTestCase):
# Check the state is unchanged
self.assertEqual(state["y"], 2)
# Check the udpated state is used
# Check the updated state is used
state["y"] = mx.array(3)
out = test_state(mx.array(1))
self.assertEqual(out.item(), 4)