fix random_utils.choice doc tests

fgnt · Dec 30, 2023 · 3e4188a · 3e4188a
1 parent 81a9b72
commit 3e4188a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/paderbox/utils/random_utils.py b/paderbox/utils/random_utils.py
@@ -453,15 +453,15 @@ def choice(*shape, events, replace=True, p=None, dtype=np.float64):
 
     Returns:
 
-    >>> x = choice()
+    >>> x = choice(events=2)
     >>> x.ndim
     0
     >>> x.dtype
     dtype('float64')
-    >>> x = choice(2, 3)
+    >>> x = choice(2, 3, events=2)
     >>> x.shape, x.dtype
     ((2, 3), dtype('float64'))
-    >>> x = choice(2, 3, dtype=np.complex128)
+    >>> x = choice(2, 3, events=2, dtype=np.complex128)
     >>> x.shape, x.dtype
     ((2, 3), dtype('complex128'))
     >>> np.random.seed(2)