scvi.distributions.ZeroInflatedNegativeBinomial.arg_constraints

ZeroInflatedNegativeBinomial.arg_constraints = {'mu': GreaterThanEq(lower_bound=0), 'theta': GreaterThanEq(lower_bound=0), 'zi_logits': Real(), 'zi_probs': HalfOpenInterval(lower_bound=0.0, upper_bound=1.0)}