From 284bbadbaee4d5885097da682fd1c119fa02989a Mon Sep 17 00:00:00 2001 From: Albert Zeyer Date: Fri, 17 Nov 2023 13:21:45 +0000 Subject: [PATCH] RF random_uniform, fix for int --- returnn/frontend/rand.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/returnn/frontend/rand.py b/returnn/frontend/rand.py index ebcf9f0d22..4a59a3bffb 100644 --- a/returnn/frontend/rand.py +++ b/returnn/frontend/rand.py @@ -234,8 +234,8 @@ def random_uniform( device: Optional[str] = None, sparse_dim: Optional[Dim] = None, feature_dim: Optional[Dim] = None, - minval: Union[int, float, Tensor] = 0.0, - maxval: Union[int, float, Tensor] = 1.0, + minval: Union[int, float, Tensor] = 0, + maxval: Union[int, float, Tensor] = 1, seed: Optional[Union[int, Sequence[int], numpy.ndarray]] = None, algorithm: Optional[str] = None, explicit_state: Optional[Tensor] = None,