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

fix test (#627)

Browse Source
This commit is contained in:
Awni Hannun 2024-02-04 16:18:03 -08:00 committed by GitHub
parent ebfd3618b0
commit 5c3ac52dd7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python/src/array.cpp
View File

@ -1110,7 +1110,8 @@ void init_array(py::module_& m) {
py::kw_only(),
"stream"_a = none,
"See :func:`abs`.")
.def("__abs__", &mlx::core::abs, "See :func:`abs`.")
.def(
"__abs__", [](const array& a) { return abs(a); }, "See :func:`abs`.")
.def(
"square",
&square,