From 99b9c1dac51c93c62718f7e8467e9637a03f5961 Mon Sep 17 00:00:00 2001
From: dc-dc-dc <dgcruz983@gmail.com>
Date: Mon, 18 Dec 2023 22:12:42 -0500
Subject: [PATCH] fix doc

---
 python/src/ops.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/src/ops.cpp b/python/src/ops.cpp
index 2189a0a10..2cf8b70dd 100644
--- a/python/src/ops.cpp
+++ b/python/src/ops.cpp
@@ -2971,7 +2971,7 @@ void init_ops(py::module_& m) {
       py::pos_only(),
       py::kw_only(),
       R"pbdoc(
-        save_safetensor(file: str, d: Dict[str, array], /, *, stream: Union[None, Stream, Device] = None)
+        save_safetensor(file: str, d: Dict[str, array], /, *)
 
         Save array(s) to a binary file in ``.safetensors`` format.