revise docstring of Multinomial

pyro-ppl · Mar 26, 2023 · 8efe57b · 8efe57b
1 parent 267f259
commit 8efe57b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/numpyro/distributions/discrete.py b/numpyro/distributions/discrete.py
@@ -638,10 +638,11 @@ def Multinomial(
     """Multinomial distribution.
 
     :param total_count: number of trials. If this is a JAX array,
-        `total_count_max` is required to specify.
+        it is required to specify `total_count_max`.
     :param probs: event probabilities
     :param logits: event log probabilities
-    :param int total_count_max: the maximum number of trials
+    :param int total_count_max: the maximum number of trials,
+        i.e. `max(total_count)`
     """
     if probs is not None:
         return MultinomialProbs(