Skip to content

Commit 50d3914

Browse files
authored
Update gumbel function signature parameters (#2868)
1 parent cacbdbf commit 50d3914

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/src/random.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -373,7 +373,7 @@ void init_random(nb::module_& parent_module) {
373373
"key"_a = nb::none(),
374374
"stream"_a = nb::none(),
375375
nb::sig(
376-
"def gumbel(shape: Sequence[int] = [], dtype: Optional[Dtype] = float32, key: Union[None, Stream, Device] = None, stream: Optional[array] = None) -> array"),
376+
"def gumbel(shape: Sequence[int] = [], dtype: Optional[Dtype] = float32, key: Optional[array] = None, stream: Union[None, Stream, Device] = None) -> array"),
377377
R"pbdoc(
378378
Sample from the standard Gumbel distribution.
379379

0 commit comments

Comments
 (0)