Randomize seed between chain batches.

PiperOrigin-RevId: 738504856

Author: lukmaz

CHANGELOG.md

@@ -23,6 +23,14 @@ To release a new version (e.g. from `1.0.0` -> `2.0.0`):
 
 ## [Unreleased]
 
+## [1.0.7] - 2025-03-19
+
+* Bump tensorflow to 2.18.
+* Bump tensorflow-probability to 0.25.
+* Bump numpy to 2.0.2.
+* Bump pandas to 2.2.2.
+* Bump scipy to 1.13.1.
+
 ## [1.0.6] - 2025-03-18
 
 * Fix issue #548: Make time coordinate regularity check less stringent.
@@ -235,4 +243,5 @@ To release a new version (e.g. from `1.0.0` -> `2.0.0`):
 [1.0.4]: https://github.com/google/meridian/releases/tag/v1.0.4
 [1.0.5]: https://github.com/google/meridian/releases/tag/v1.0.5
 [1.0.6]: https://github.com/google/meridian/releases/tag/v1.0.6
-[Unreleased]: https://github.com/google/meridian/compare/v1.0.6...HEAD
+[1.0.7]: https://github.com/google/meridian/releases/tag/v1.0.7
+[Unreleased]: https://github.com/google/meridian/compare/v1.0.7...HEAD

README.md

@@ -151,7 +151,7 @@ To cite this repository:
   author = {Google Meridian Marketing Mix Modeling Team},
   title = {Meridian: Marketing Mix Modeling},
   url = {https://github.com/google/meridian},
-  version = {1.0.6},
+  version = {1.0.7},
   year = {2025},
 }
 ```

demo/Meridian_Getting_Started.ipynb

@@ -335,7 +335,7 @@
       "source": [
         "%%time\n",
         "mmm.sample_prior(500)\n",
-        "mmm.sample_posterior(n_chains=7, n_adapt=500, n_burnin=500, n_keep=1000)"
+        "mmm.sample_posterior(n_chains=7, n_adapt=500, n_burnin=500, n_keep=1000, seed=1)"
       ]
     },
     {

meridian/__init__.py

@@ -14,7 +14,7 @@
 
 """Meridian API."""
 
-__version__ = "1.0.6"
+__version__ = "1.0.7"
 
 
 from meridian import analysis

meridian/model/posterior_sampler.py

@@ -457,7 +457,7 @@ def __call__(
         [ResourceExhaustedError when running Meridian.sample_posterior]
         (https://developers.google.com/meridian/docs/advanced-modeling/model-debugging#gpu-oom-error).
     """
-    seed = tfp.random.sanitize_seed(seed) if seed else None
+    seed = tfp.random.sanitize_seed(seed or 0)
     n_chains_list = [n_chains] if isinstance(n_chains, int) else n_chains
     total_chains = np.sum(n_chains_list)
 
@@ -480,6 +480,7 @@ def __call__(
             seed=seed,
             **pins,
         )
+        seed = [x + 1 for x in seed]
       except tf.errors.ResourceExhaustedError as error:
         raise MCMCOOMError(
             "ERROR: Out of memory. Try reducing `n_keep` or pass a list of"

pyproject.toml

@@ -30,12 +30,12 @@ dependencies = [
   "altair >= 5",
   "immutabledict",
   "joblib",
-  "numpy >= 1.26, < 2",
-  "pandas >= 2.2, < 3",
-  "scipy >= 1.12.0, < 1.13",
-  "tensorflow >= 2.16, < 2.17",
-  "tensorflow-probability >= 0.24, < 0.25",
-  "tf-keras >= 2.16, < 2.17",
+  "numpy >= 2.0.2, < 3",
+  "pandas >= 2.2.2, < 3",
+  "scipy >= 1.13.1, < 2",
+  "tensorflow >= 2.18, < 2.19",
+  "tensorflow-probability >= 0.25, < 0.26",
+  "tf-keras >= 2.18, < 2.19",
   "xarray",
 ]
 
@@ -65,7 +65,7 @@ colab = [
 # GPU deps
 # Installed through `pip install -e .[and-cuda]`
 and-cuda = [
-  "tensorflow[and-cuda] >= 2.16, < 2.17",
+  "tensorflow[and-cuda] >= 2.18, < 2.19",
 ]
 
 [tool.pyink]