delete probs' default value/ fix sample/rsample's param

PaddlePaddle · Apr 21, 2023 · c3cdd69 · c3cdd69
1 parent 7439580
commit c3cdd69
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/paddle/distribution/geometric.py b/python/paddle/distribution/geometric.py
@@ -64,7 +64,7 @@ class Geometric(distribution.Distribution):
             #        [1.41421354])
     """
 
-    def __init__(self, probs=None):
+    def __init__(self, probs):
         if isinstance(probs, (numbers.Real, paddle.Tensor, framework.Variable)):
             if isinstance(probs, numbers.Real):
                 probs = paddle.full(
@@ -187,7 +187,7 @@ def sample(self, shape=()):
         """Sample from Geometric distribution with sample shape.
 
         Args:
-            shape (Sequence[int], optional): Sample shape.
+            shape (tuple(int)): Sample shape.
 
         Returns:
             Sampled data with shape `sample_shape` + `batch_shape` + `event_shape`.
@@ -214,7 +214,7 @@ def rsample(self, shape=()):
         """Generate samples of the specified shape.
 
         Args:
-            shape(tuple): The shape of generated samples.
+            shape(tuple(int)): The shape of generated samples.
 
         Returns:
             Tensor: A sample tensor that fits the Geometric distribution.