make python array SupportsAbs conform (like numpy) (#624)

2025-12-14 17:12:49 +08:00 · 2024-02-04 18:31:02 +01:00
parent 9852af1a19
commit 4fd2fb84a6
2 changed files with 3 additions and 0 deletions
--- a/python/src/array.cpp
+++ b/python/src/array.cpp
@@ -1110,6 +1110,7 @@ void init_array(py::module_& m) {
          py::kw_only(),
          "stream"_a = none,
          "See :func:`abs`.")
+      .def("__abs__", &mlx::core::abs, "See :func:`abs`.")
      .def(
          "square",
          &square,