Skip to content

Commit

Permalink
Revert "set a mock idata posterior"
Browse files Browse the repository at this point in the history
This reverts commit 2408f61.
  • Loading branch information
wd60622 authored and juanitorduz committed Feb 9, 2024
1 parent 315e0ec commit a224cfe
Showing 1 changed file with 8 additions and 13 deletions.
21 changes: 8 additions & 13 deletions tests/mmm/test_delayed_saturated_mmm.py
Original file line number Diff line number Diff line change
Expand Up @@ -106,29 +106,24 @@ def mmm_with_fourier_features() -> DelayedSaturatedMMM:
)


def mmm_with_prior_as_posterior(
mmm: DelayedSaturatedMMM, X: pd.DataFrame
) -> DelayedSaturatedMMM:
mmm.sample_prior_predictive(X, samples=2000, extend_idata=True, random_seed=rng)
mmm.idata.add_groups({"posterior": mmm.idata.prior})

return mmm


@pytest.fixture(scope="module")
def mmm_fitted(
mmm: DelayedSaturatedMMM,
toy_X: pd.DataFrame,
mmm: DelayedSaturatedMMM, toy_X: pd.DataFrame, toy_y: pd.Series
) -> DelayedSaturatedMMM:
return mmm_with_prior_as_posterior(mmm, toy_X)
mmm.fit(X=toy_X, y=toy_y, target_accept=0.8, draws=3, chains=2)
return mmm


@pytest.fixture(scope="module")
def mmm_fitted_with_fourier_features(
mmm_with_fourier_features: DelayedSaturatedMMM,
toy_X: pd.DataFrame,
toy_y: pd.Series,
) -> DelayedSaturatedMMM:
return mmm_with_prior_as_posterior(mmm_with_fourier_features, toy_X)
mmm_with_fourier_features.fit(
X=toy_X, y=toy_y, target_accept=0.8, draws=3, chains=2
)
return mmm_with_fourier_features


class TestDelayedSaturatedMMM:
Expand Down

0 comments on commit a224cfe

Please sign in to comment.