diff --git a/numpyro/distributions/continuous.py b/numpyro/distributions/continuous.py index 262295b3e..231d37fb9 100644 --- a/numpyro/distributions/continuous.py +++ b/numpyro/distributions/continuous.py @@ -2654,7 +2654,7 @@ class Wishart(TransformedDistribution): def __init__( self, - concentration=None, + concentration, scale_matrix=None, rate_matrix=None, scale_tril=None, @@ -2748,7 +2748,7 @@ class WishartCholesky(Distribution): def __init__( self, - concentration=None, + concentration, scale_matrix=None, rate_matrix=None, scale_tril=None,