test for marketig version

pymc-labs · Aug 13, 2024 · 8003ddf · 8003ddf
1 parent 3aa6c8f
commit 8003ddf
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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"])