fix transformed distribution validate samples (#1465)

pyro-ppl · Aug 9, 2022 · f48e341 · f48e341
1 parent be1eb6b
commit f48e341
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/numpyro/distributions/util.py b/numpyro/distributions/util.py
@@ -624,7 +624,7 @@ def __get__(self, instance, obj_type=None):
 
 def validate_sample(log_prob_fn):
     def wrapper(self, *args, **kwargs):
-        log_prob = log_prob_fn(self, *args, *kwargs)
+        log_prob = log_prob_fn(self, *args, **kwargs)
         if self._validate_args:
             value = kwargs["value"] if "value" in kwargs else args[0]
             mask = self._validate_sample(value)