diff --git a/mlx/ops.h b/mlx/ops.h index a99465f3e..c888c80cd 100644 --- a/mlx/ops.h +++ b/mlx/ops.h @@ -214,7 +214,7 @@ array concatenate(const std::vector& arrays, StreamOrDevice s = {}); array stack(const std::vector& arrays, int axis, StreamOrDevice s = {}); array stack(const std::vector& arrays, StreamOrDevice s = {}); -/** Repeate an array along an axis. */ +/** Repeat an array along an axis. */ array repeat(const array& arr, int repeats, int axis, StreamOrDevice s = {}); array repeat(const array& arr, int repeats, StreamOrDevice s = {}); diff --git a/python/src/ops.cpp b/python/src/ops.cpp index cab5f28c6..5e3d0534c 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -2426,7 +2426,7 @@ void init_ops(py::module_& m) { R"pbdoc( repeat(array: array, repeats: int, axis: Optional[int] = None, *, stream: Union[None, Stream, Device] = None) -> array - Repeate an array along a specified axis. + Repeat an array along a specified axis. Args: array (array): Input array.