RF random_uniform, fix for int

rwth-i6 · Nov 17, 2023 · 284bbad · 284bbad
1 parent d0e17fd
commit 284bbad
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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,