scvi.distributions.NegativeBinomialMixture.arg_constraints

NegativeBinomialMixture.arg_constraints = {'mixture_logits': Real(), 'mixture_probs': HalfOpenInterval(lower_bound=0.0, upper_bound=1.0), 'mu1': GreaterThanEq(lower_bound=0), 'mu2': GreaterThanEq(lower_bound=0), 'theta1': GreaterThanEq(lower_bound=0)}