Fix GammaPoisson distribution from ignoring validate_args param (#3253)

pyro-ppl · Aug 2, 2023 · 25fd264 · 25fd264
1 parent 7944968
commit 25fd264
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyro/distributions/conjugate.py b/pyro/distributions/conjugate.py
@@ -249,7 +249,8 @@ class GammaPoisson(TorchDistribution):
 
     def __init__(self, concentration, rate, validate_args=None):
         concentration, rate = broadcast_all(concentration, rate)
-        self._gamma = Gamma(concentration, rate)
+        self._gamma = Gamma(concentration, rate, validate_args=False)
+        self._gamma._validate_args = validate_args
         super().__init__(self._gamma._batch_shape, validate_args=validate_args)
 
     @property