From 8003ddfd6068dacaff6c2d3314489e067e8090ac Mon Sep 17 00:00:00 2001 From: Will Dean Date: Wed, 14 Aug 2024 01:56:19 +0200 Subject: [PATCH] test for marketig version --- tests/test_mlflow.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tests/test_mlflow.py b/tests/test_mlflow.py index 5a7de7dc..a5297510 100644 --- a/tests/test_mlflow.py +++ b/tests/test_mlflow.py @@ -32,6 +32,7 @@ log_sample_diagnostics, ) from pymc_marketing.mmm import MMM, GeometricAdstock, LogisticSaturation +from pymc_marketing.version import __version__ seed = sum(map(ord, "mlflow-with-pymc")) rng = np.random.default_rng(seed) @@ -235,6 +236,8 @@ def param_checks(params, draws: int, chains: int, tune: int, nuts_sampler: str) if nuts_sampler not in ["numpyro", "nutpie", "blackjax"]: assert params["tuning_steps"] == str(tune) + assert params["pymc_marketing_version"] == __version__ + other_keys = ["pymc_version"] if nuts_sampler not in ["numpyro", "blackjax"]: other_keys.extend(["inference_library_version"])