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

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.
2025-09-01 12:49:44 +08:00 · 2024-05-15 17:25:25 -05:00
parent 358e1fd6ab
commit c417e42116
1 changed files with 1 additions and 1 deletions
--- a/python/src/ops.cpp
+++ b/python/src/ops.cpp
@@ -2527,7 +2527,7 @@ void init_ops(nb::module_& m) {
      },
      nb::arg(),
      "kth"_a,
-      "axis"_a = -1,
+      "axis"_a.none() = -1,
      nb::kw_only(),
      "stream"_a = nb::none(),
      nb::sig(