diff --git a/pts/distributions/implicit_quantile.py b/pts/distributions/implicit_quantile.py index c950bdf..5fdf1ab 100644 --- a/pts/distributions/implicit_quantile.py +++ b/pts/distributions/implicit_quantile.py @@ -3,6 +3,8 @@ from torch.distributions import Distribution, TransformedDistribution, AffineTra class ImplicitQuantile(Distribution): + arg_constraints = {} + def __init__( self, implicit_quantile_function,