From c417e421169011d01888a495af2d6dde6b9e44c5 Mon Sep 17 00:00:00 2001
From: Jacket <44538064+PRESIDENT810@users.noreply.github.com>
Date: Wed, 15 May 2024 17:25:25 -0500
Subject: [PATCH] [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.
---
 python/src/ops.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/src/ops.cpp b/python/src/ops.cpp
index 17f7ed897..82c141e31 100644
--- 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(