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

[Fix] minor typo in default argument for argpartition's "axis" parameter (#1125)

Browse Source 
According to the document, argpartition's axis parameter can be None, but due to a previous typo it can't really accepts a None value.
This commit is contained in:
Jacket 2024-05-15 17:25:25 -05:00 committed by GitHub
parent 358e1fd6ab
commit c417e42116
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
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/src/ops.cpp
View File

@ -2527,7 +2527,7 @@ void init_ops(nb::module_& m) {
}, },
nb::arg(), nb::arg(),
"kth"_a, "kth"_a,
"axis"_a = -1, "axis"_a.none() = -1,
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(