scvi.distributions.NegativeBinomial.arg_constraints

NegativeBinomial.arg_constraints = {'mu': GreaterThanEq(lower_bound=0), 'theta': GreaterThanEq(lower_bound=0)}