def __init__(self, rate, *, gate=None, gate_logits=None, validate_args=None): base_dist = Poisson(rate=rate, validate_args=False) base_dist._validate_args = validate_args super().__init__( base_dist, gate=gate, gate_logits=gate_logits, validate_args=validate_args )
def __init__(self, gate, rate, validate_args=None): base_dist = Poisson(rate=rate, validate_args=False) base_dist._validate_args = validate_args super().__init__(gate, base_dist, validate_args=validate_args)