L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update pre-commit hooks and versions for clang-format, black, and isort (#3059)

Browse Source
This commit is contained in:
Nripesh Niketan
2026-01-26 14:57:04 +00:00
committed by GitHub
parent 5bd99dd5ec
commit b6aa03e5b8
37 changed files with 376 additions and 335 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/src/ops.cpp
+12 -8
View File
@@ -53,8 +53,9 @@ void init_ops(nb::module_& m) {
"shape"_a,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig("def reshape(a: array, /, shape: Sequence[int], *, stream: "
"Union[None, Stream, Device] = None) -> array"),
nb::sig(
"def reshape(a: array, /, shape: Sequence[int], *, stream: "
"Union[None, Stream, Device] = None) -> array"),
R"pbdoc(
Reshape an array while preserving the size.
@@ -80,8 +81,9 @@ void init_ops(nb::module_& m) {
"end_axis"_a = -1,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig("def flatten(a: array, /, start_axis: int = 0, end_axis: int = "
"-1, *, stream: Union[None, Stream, Device] = None) -> array"),
nb::sig(
"def flatten(a: array, /, start_axis: int = 0, end_axis: int = "
"-1, *, stream: Union[None, Stream, Device] = None) -> array"),
R"pbdoc(
Flatten an array.
@@ -182,8 +184,9 @@ void init_ops(nb::module_& m) {
"axis"_a,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig("def expand_dims(a: array, /, axis: Union[int, Sequence[int]], "
"*, stream: Union[None, Stream, Device] = None) -> array"),
nb::sig(
"def expand_dims(a: array, /, axis: Union[int, Sequence[int]], "
"*, stream: Union[None, Stream, Device] = None) -> array"),
R"pbdoc(
Add a size one dimension at the given axis.
@@ -1675,8 +1678,9 @@ void init_ops(nb::module_& m) {
},
nb::arg(),
"dtype"_a = nb::none(),
nb::sig("def asarray(a: Union[scalar, array, Sequence], dtype: "
"Optional[Dtype] = None) -> array"),
nb::sig(
"def asarray(a: Union[scalar, array, Sequence], dtype: "
"Optional[Dtype] = None) -> array"),
R"pbdoc(
Convert the input to an array.