From 56a4eaed72febffd5e3be00ebfe78da04a67ffde Mon Sep 17 00:00:00 2001
From: Cheng <git@zcbenz.com>
Date: Tue, 14 May 2024 21:50:16 +0800
Subject: [PATCH] Pass missing stream arg in array.flatten (#1111)

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

diff --git a/python/src/array.cpp b/python/src/array.cpp
index 243328dbe..943ce2a5a 100644
--- a/python/src/array.cpp
+++ b/python/src/array.cpp
@@ -1139,7 +1139,7 @@ void init_array(nb::module_& m) {
              int start_axis,
              int end_axis,
              const StreamOrDevice& s) {
-            return flatten(a, start_axis, end_axis);
+            return flatten(a, start_axis, end_axis, s);
           },
           "start_axis"_a = 0,
           "end_axis"_a = -1,