@Appender(ZeroInflatedPoisson.get_distribution.__doc__)
def get_distribution(self, params, exog=None, exog_infl=None, exposure=None, offset=None):
    p = self.model_main.parameterization
    mu = self.predict(params, exog=exog, exog_infl=exog_infl, exposure=exposure, offset=offset, which='mean-main')
    w = self.predict(params, exog=exog, exog_infl=exog_infl, exposure=exposure, offset=offset, which='prob-main')
    distr = self.distribution(mu[:, None], params[(- 1)], p, (1 - w[:, None]))
    return distr
@Appender(ZeroInflatedPoisson.get_distribution.__doc__)
def get_distribution(self, params, exog=None, exog_infl=None, exposure=None, offset=None):
    p = self.model_main.parameterization
    mu = self.predict(params, exog=exog, exog_infl=exog_infl, exposure=exposure, offset=offset, which='mean-main')
    w = self.predict(params, exog=exog, exog_infl=exog_infl, exposure=exposure, offset=offset, which='prob-main')
    distr = self.distribution(mu[:, None], params[(- 1)], p, (1 - w[:, None]))
    return distr