Skip to content

Commit f48e341

Browse files
authored
fix transformed distribution validate samples (#1465)
1 parent be1eb6b commit f48e341

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

numpyro/distributions/util.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -624,7 +624,7 @@ def __get__(self, instance, obj_type=None):
624624

625625
def validate_sample(log_prob_fn):
626626
def wrapper(self, *args, **kwargs):
627-
log_prob = log_prob_fn(self, *args, *kwargs)
627+
log_prob = log_prob_fn(self, *args, **kwargs)
628628
if self._validate_args:
629629
value = kwargs["value"] if "value" in kwargs else args[0]
630630
mask = self._validate_sample(value)

0 commit comments

Comments
 (0)