From 51322b90e3804679c0bc7acc1784cc63648a3a46 Mon Sep 17 00:00:00 2001 From: Awni Hannun Date: Sat, 23 Dec 2023 06:42:41 -0800 Subject: [PATCH] comment --- python/src/ops.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/src/ops.cpp b/python/src/ops.cpp index db405b127..8a65b1053 100644 --- a/python/src/ops.cpp +++ b/python/src/ops.cpp @@ -2929,12 +2929,12 @@ void init_ops(py::module_& m) { "load", &mlx_load_helper, "file"_a, - "format"_a = none, py::pos_only(), + "format"_a = none, py::kw_only(), "stream"_a = none, R"pbdoc( - load(file: str, format: Optional[str] = None, /, *, stream: Union[None, Stream, Device] = None) -> Union[array, Dict[str, array]] + load(file: str, /, format: Optional[str] = None, *, stream: Union[None, Stream, Device] = None) -> Union[array, Dict[str, array]] Load array(s) from a binary file in ``.npy``, ``.npz``, or ``.safetensors`` format.