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

Fix test

Browse Source
This commit is contained in:
Angelos Katharopoulos 2025-01-14 17:14:44 -08:00
parent 2e08e8b96c
commit 5278143c1b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
llms/tests/test_finetune.py
View File

@ -21,7 +21,7 @@ from mlx_lm.tuner.utils import build_schedule
@contextmanager @contextmanager
def swapped_with_identity(obj, func): def swapped_with_identity(obj, func):
old_func = getattr(obj, func) old_func = getattr(obj, func)
setattr(obj, func, lambda x: x) setattr(obj, func, lambda x, **kwargs: x)
yield yield
setattr(obj, func, old_func) setattr(obj, func, old_func)