From 50d3914c67204d9629e6599269f3e663e1b281ec Mon Sep 17 00:00:00 2001
From: "Tian En \"TianHeng" <tianenchong@gmail.com>
Date: Wed, 3 Dec 2025 23:37:35 +0000
Subject: [PATCH] Update gumbel function signature parameters (#2868)

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

diff --git a/python/src/random.cpp b/python/src/random.cpp
index 0972c2cc3..c832c5a9e 100644
--- a/python/src/random.cpp
+++ b/python/src/random.cpp
@@ -373,7 +373,7 @@ void init_random(nb::module_& parent_module) {
       "key"_a = nb::none(),
       "stream"_a = nb::none(),
       nb::sig(
-          "def gumbel(shape: Sequence[int] = [], dtype: Optional[Dtype] = float32, key: Union[None, Stream, Device] = None, stream: Optional[array] = None) -> array"),
+          "def gumbel(shape: Sequence[int] = [], dtype: Optional[Dtype] = float32, key: Optional[array] = None, stream: Union[None, Stream, Device] = None) -> array"),
       R"pbdoc(
         Sample from the standard Gumbel distribution.