google-meridian 1.3.1__tar.gz → 1.3.2__tar.gz

google_meridian-1.3.2/MANIFEST.in

@@ -0,0 +1,2 @@
+global-exclude *_test.py
+include meridian/templates/*

{google_meridian-1.3.1/google_meridian.egg-info → google_meridian-1.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-meridian
-Version: 1.3.1
+Version: 1.3.2
 Summary: Google's open source mixed marketing model library, helps you understand your return on investment and direct your ad spend with confidence. 
 Author-email: The Meridian Authors <no-reply@google.com>
 Project-URL: homepage, https://github.com/google/meridian
@@ -25,9 +25,9 @@ Requires-Dist: pandas<3,>=2.2.2
 Requires-Dist: patsy<1,>=0.5.3
 Requires-Dist: scipy<2,>=1.13.1
 Requires-Dist: statsmodels>=0.14.5
-Requires-Dist: tensorflow<2.19,>=2.18
+Requires-Dist: tensorflow<2.21,>=2.18
 Requires-Dist: tensorflow-probability<0.26,>=0.25
-Requires-Dist: tf-keras<2.19,>=2.18
+Requires-Dist: tf-keras<2.21,>=2.18
 Requires-Dist: xarray
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
@@ -38,12 +38,12 @@ Provides-Extra: colab
 Requires-Dist: psutil; extra == "colab"
 Requires-Dist: python-calamine; extra == "colab"
 Provides-Extra: and-cuda
-Requires-Dist: tensorflow[and-cuda]<2.19,>=2.18; extra == "and-cuda"
+Requires-Dist: tensorflow[and-cuda]<2.21,>=2.18; extra == "and-cuda"
 Provides-Extra: mlflow
 Requires-Dist: mlflow; extra == "mlflow"
 Provides-Extra: jax
-Requires-Dist: jax==0.4.26; extra == "jax"
-Requires-Dist: jaxlib==0.4.26; extra == "jax"
+Requires-Dist: jax==0.5.3; extra == "jax"
+Requires-Dist: jaxlib==0.5.3; extra == "jax"
 Requires-Dist: tensorflow-probability[substrates-jax]==0.25.0; extra == "jax"
 Provides-Extra: schema
 Requires-Dist: mmm-proto-schema; extra == "schema"
@@ -203,7 +203,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.3.1},
+  version = {1.3.2},
   year = {2025},
 }
 ```

{google_meridian-1.3.1 → google_meridian-1.3.2}/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.3.1},
+  version = {1.3.2},
   year = {2025},
 }
 ```

{google_meridian-1.3.1 → google_meridian-1.3.2/google_meridian.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-meridian
-Version: 1.3.1
+Version: 1.3.2
 Summary: Google's open source mixed marketing model library, helps you understand your return on investment and direct your ad spend with confidence. 
 Author-email: The Meridian Authors <no-reply@google.com>
 Project-URL: homepage, https://github.com/google/meridian
@@ -25,9 +25,9 @@ Requires-Dist: pandas<3,>=2.2.2
 Requires-Dist: patsy<1,>=0.5.3
 Requires-Dist: scipy<2,>=1.13.1
 Requires-Dist: statsmodels>=0.14.5
-Requires-Dist: tensorflow<2.19,>=2.18
+Requires-Dist: tensorflow<2.21,>=2.18
 Requires-Dist: tensorflow-probability<0.26,>=0.25
-Requires-Dist: tf-keras<2.19,>=2.18
+Requires-Dist: tf-keras<2.21,>=2.18
 Requires-Dist: xarray
 Provides-Extra: dev
 Requires-Dist: pytest>=8.0.0; extra == "dev"
@@ -38,12 +38,12 @@ Provides-Extra: colab
 Requires-Dist: psutil; extra == "colab"
 Requires-Dist: python-calamine; extra == "colab"
 Provides-Extra: and-cuda
-Requires-Dist: tensorflow[and-cuda]<2.19,>=2.18; extra == "and-cuda"
+Requires-Dist: tensorflow[and-cuda]<2.21,>=2.18; extra == "and-cuda"
 Provides-Extra: mlflow
 Requires-Dist: mlflow; extra == "mlflow"
 Provides-Extra: jax
-Requires-Dist: jax==0.4.26; extra == "jax"
-Requires-Dist: jaxlib==0.4.26; extra == "jax"
+Requires-Dist: jax==0.5.3; extra == "jax"
+Requires-Dist: jaxlib==0.5.3; extra == "jax"
 Requires-Dist: tensorflow-probability[substrates-jax]==0.25.0; extra == "jax"
 Provides-Extra: schema
 Requires-Dist: mmm-proto-schema; extra == "schema"
@@ -203,7 +203,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.3.1},
+  version = {1.3.2},
   year = {2025},
 }
 ```

{google_meridian-1.3.1 → google_meridian-1.3.2}/google_meridian.egg-info/SOURCES.txt

@@ -13,7 +13,6 @@ meridian/constants.py
 meridian/version.py
 meridian/analysis/__init__.py
 meridian/analysis/analyzer.py
-meridian/analysis/formatter.py
 meridian/analysis/optimizer.py
 meridian/analysis/summarizer.py
 meridian/analysis/summary_text.py
@@ -25,14 +24,6 @@ meridian/analysis/review/configs.py
 meridian/analysis/review/constants.py
 meridian/analysis/review/results.py
 meridian/analysis/review/reviewer.py
-meridian/analysis/templates/card.html.jinja
-meridian/analysis/templates/chart.html.jinja
-meridian/analysis/templates/chips.html.jinja
-meridian/analysis/templates/insights.html.jinja
-meridian/analysis/templates/stats.html.jinja
-meridian/analysis/templates/style.scss
-meridian/analysis/templates/summary.html.jinja
-meridian/analysis/templates/table.html.jinja
 meridian/backend/__init__.py
 meridian/backend/config.py
 meridian/backend/test_utils.py
@@ -63,7 +54,16 @@ meridian/model/eda/constants.py
 meridian/model/eda/eda_engine.py
 meridian/model/eda/eda_outcome.py
 meridian/model/eda/eda_spec.py
-meridian/model/eda/meridian_eda.py
+meridian/templates/card.html.jinja
+meridian/templates/chart.html.jinja
+meridian/templates/chips.html.jinja
+meridian/templates/formatter.py
+meridian/templates/formatter_test.py
+meridian/templates/insights.html.jinja
+meridian/templates/stats.html.jinja
+meridian/templates/style.scss
+meridian/templates/summary.html.jinja
+meridian/templates/table.html.jinja
 schema/__init__.py
 schema/serde/__init__.py
 schema/serde/constants.py

{google_meridian-1.3.1 → google_meridian-1.3.2}/google_meridian.egg-info/requires.txt

@@ -8,13 +8,13 @@ pandas<3,>=2.2.2
 patsy<1,>=0.5.3
 scipy<2,>=1.13.1
 statsmodels>=0.14.5
-tensorflow<2.19,>=2.18
+tensorflow<2.21,>=2.18
 tensorflow-probability<0.26,>=0.25
-tf-keras<2.19,>=2.18
+tf-keras<2.21,>=2.18
 xarray
 
 [and-cuda]
-tensorflow[and-cuda]<2.19,>=2.18
+tensorflow[and-cuda]<2.21,>=2.18
 
 [colab]
 psutil
@@ -27,8 +27,8 @@ pylint>=2.6.0
 pyink
 
 [jax]
-jax==0.4.26
-jaxlib==0.4.26
+jax==0.5.3
+jaxlib==0.5.3
 tensorflow-probability[substrates-jax]==0.25.0
 
 [mlflow]

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/__init__.py

@@ -15,9 +15,8 @@
 """Meridian analysis API for trained models."""
 
 from meridian.analysis import analyzer
-from meridian.analysis import formatter
 from meridian.analysis import optimizer
 from meridian.analysis import review
 from meridian.analysis import summarizer
 from meridian.analysis import visualizer
-
+from meridian.templates import formatter

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/analyzer.py

@@ -53,7 +53,6 @@ def _validate_non_media_baseline_values_numbers(
       )
 
 
-# TODO: Refactor the related unit tests to be under DataTensors.
 @dataclasses.dataclass
 class DataTensors(backend.ExtensionType):
   """Container for data variable arguments of Analyzer methods.

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/optimizer.py

@@ -27,10 +27,10 @@ import jinja2
 from meridian import backend
 from meridian import constants as c
 from meridian.analysis import analyzer as analyzer_module
-from meridian.analysis import formatter
 from meridian.analysis import summary_text
 from meridian.data import time_coordinates as tc
 from meridian.model import model
+from meridian.templates import formatter
 import numpy as np
 import pandas as pd
 import xarray as xr
@@ -1174,10 +1174,12 @@ class OptimizationResults:
       diff = self.optimized_data.total_cpik - self.nonoptimized_data.total_cpik
       non_optimized_performance_title = summary_text.NON_OPTIMIZED_CPIK_LABEL
       non_optimized_performance_stat = (
-          f'${self.nonoptimized_data.total_cpik:.2f}'
+          f'{currency}{self.nonoptimized_data.total_cpik:.2f}'
       )
       optimized_performance_title = summary_text.OPTIMIZED_CPIK_LABEL
-      optimized_performance_stat = f'${self.optimized_data.total_cpik:.2f}'
+      optimized_performance_stat = (
+          f'{currency}{self.optimized_data.total_cpik:.2f}'
+      )
       optimized_performance_diff = formatter.compact_number(diff, 2, currency)
     non_optimized_performance = formatter.StatsSpec(
         title=non_optimized_performance_title,

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/review/checks.py

@@ -28,6 +28,7 @@ from meridian.analysis.review import constants as review_constants
 from meridian.analysis.review import results
 from meridian.model import model
 import numpy as np
+import pandas as pd
 
 ConfigType = TypeVar("ConfigType", bound=configs.BaseConfig)
 ResultType = TypeVar("ResultType", bound=results.CheckResult)
@@ -207,6 +208,51 @@ class BayesianPPPCheck(
 # ==============================================================================
 # Check: Goodness of Fit
 # ==============================================================================
+def _set_details_from_gof_dataframe(
+    details: dict[str, float],
+    gof_df: pd.DataFrame,
+    geo_granularity: str,
+    suffix: str | None = None,
+) -> None:
+  """Sets the `details` variable of the GoodnessOfFitCheckResult.
+
+  This method takes a DataFrame containing goodness of fit metrics and pivots it
+  to a Series, which is then added to the `details` variable of the
+  `GoodnessOfFitCheckResult`.
+
+  Args:
+    details: A dictionary to store the goodness of fit metrics in.
+    gof_df: A DataFrame containing predictive accuracy of the whole data (if
+      holdout set is not used) of filtered to a single evaluation set ("all",
+      "train", or "test").
+    geo_granularity: The geo granularity of the data ("geo" or "national").
+    suffix: A suffix to add to the metric names (e.g., "all", "train", "test").
+      If None, the metrics are added without a suffix.
+  """
+  gof_metrics_pivoted = gof_df.pivot(
+      index=constants.GEO_GRANULARITY,
+      columns=constants.METRIC,
+      values=constants.VALUE,
+  )
+  gof_metrics_series = gof_metrics_pivoted.loc[geo_granularity]
+  if suffix is not None:
+    details[f"{review_constants.R_SQUARED}_{suffix}"] = gof_metrics_series[
+        constants.R_SQUARED
+    ]
+    details[f"{review_constants.MAPE}_{suffix}"] = gof_metrics_series[
+        constants.MAPE
+    ]
+    details[f"{review_constants.WMAPE}_{suffix}"] = gof_metrics_series[
+        constants.WMAPE
+    ]
+  else:
+    details[review_constants.R_SQUARED] = gof_metrics_series[
+        constants.R_SQUARED
+    ]
+    details[review_constants.MAPE] = gof_metrics_series[constants.MAPE]
+    details[review_constants.WMAPE] = gof_metrics_series[constants.WMAPE]
+
+
 class GoodnessOfFitCheck(
     BaseCheck[configs.GoodnessOfFitConfig, results.GoodnessOfFitCheckResult]
 ):
@@ -221,38 +267,43 @@ class GoodnessOfFitCheck(
     )
 
     gof_metrics = gof_df[gof_df[constants.GEO_GRANULARITY] == geo_granularity]
-    if constants.EVALUATION_SET_VAR in gof_df.columns:
-      gof_metrics = gof_metrics[
-          gof_metrics[constants.EVALUATION_SET_VAR] == constants.ALL_DATA
-      ]
-
-    gof_metrics_pivoted = gof_metrics.pivot(
-        index=constants.GEO_GRANULARITY,
-        columns=constants.METRIC,
-        values=constants.VALUE,
-    )
-    gof_metrics_series = gof_metrics_pivoted.loc[geo_granularity]
-
-    r_squared = gof_metrics_series[constants.R_SQUARED]
-    mape = gof_metrics_series[constants.MAPE]
-    wmape = gof_metrics_series[constants.WMAPE]
-
-    details = {
-        review_constants.R_SQUARED: r_squared,
-        review_constants.MAPE: mape,
-        review_constants.WMAPE: wmape,
-    }
-
-    if r_squared > 0:
-      return results.GoodnessOfFitCheckResult(
-          case=results.GoodnessOfFitCases.PASS,
-          details=details,
-      )
-    else:  # r_squared <= 0
-      return results.GoodnessOfFitCheckResult(
-          case=results.GoodnessOfFitCases.REVIEW,
+    is_holdout = constants.EVALUATION_SET_VAR in gof_df.columns
+
+    details = {}
+    case = results.GoodnessOfFitCases.PASS
+
+    if is_holdout:
+      for evaluation_set, suffix in [
+          (constants.ALL_DATA, review_constants.ALL_SUFFIX),
+          (constants.TRAIN, review_constants.TRAIN_SUFFIX),
+          (constants.TEST, review_constants.TEST_SUFFIX),
+      ]:
+        set_metrics = gof_metrics[
+            gof_metrics[constants.EVALUATION_SET_VAR] == evaluation_set
+        ]
+        _set_details_from_gof_dataframe(
+            details=details,
+            gof_df=set_metrics,
+            geo_granularity=geo_granularity,
+            suffix=suffix,
+        )
+        if details[f"{review_constants.R_SQUARED}_{suffix}"] <= 0:
+          case = results.GoodnessOfFitCases.REVIEW
+    else:
+      _set_details_from_gof_dataframe(
           details=details,
+          gof_df=gof_metrics,
+          geo_granularity=geo_granularity,
+          suffix=None,
       )
+      if details[review_constants.R_SQUARED] <= 0:
+        case = results.GoodnessOfFitCases.REVIEW
+
+    return results.GoodnessOfFitCheckResult(
+        case=case,
+        details=details,
+        is_holdout=is_holdout,
+    )
 
 
 # ==============================================================================

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/review/constants.py

@@ -32,6 +32,10 @@ NEGATIVE_BASELINE_PROB_REVIEW_THRESHOLD = (
 R_SQUARED = "r_squared"
 MAPE = "mape"
 WMAPE = "wmape"
+ALL_SUFFIX = "all"
+TRAIN_SUFFIX = "train"
+TEST_SUFFIX = "test"
+EVALUATION_SET_SUFFIXES = (ALL_SUFFIX, TRAIN_SUFFIX, TEST_SUFFIX)
 MEAN = "mean"
 VARIANCE = "variance"
 MEDIAN = "median"

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/review/results.py

@@ -319,18 +319,12 @@ class GoodnessOfFitCases(ModelCheckCase, enum.Enum):
 
   PASS = (
       Status.PASS,
-      (
-          "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE ="
-          " {wmape:.4f}."
-      ),
+      "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE = {wmape:.4f}",
       _GOODNESS_OF_FIT_PASS_RECOMMENDATION,
   )
   REVIEW = (
       Status.REVIEW,
-      (
-          "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE ="
-          " {wmape:.4f}."
-      ),
+      "R-squared = {r_squared:.4f}, MAPE = {mape:.4f}, and wMAPE = {wmape:.4f}",
       _GOODNESS_OF_FIT_REVIEW_RECOMMENDATION,
   )
 
@@ -348,9 +342,28 @@ class GoodnessOfFitCheckResult(CheckResult):
   """The immutable result of the Goodness of Fit Check."""
 
   case: GoodnessOfFitCases
+  is_holdout: bool = False
 
   def __post_init__(self):
-    if any(
+    if self.is_holdout:
+      required_keys = []
+      for suffix in [
+          constants.ALL_SUFFIX,
+          constants.TRAIN_SUFFIX,
+          constants.TEST_SUFFIX,
+      ]:
+        required_keys.extend([
+            f"{constants.R_SQUARED}_{suffix}",
+            f"{constants.MAPE}_{suffix}",
+            f"{constants.WMAPE}_{suffix}",
+        ])
+      if any(key not in self.details for key in required_keys):
+        raise ValueError(
+            "The message template is missing required formatting arguments for"
+            f" holdout case. Required keys: {required_keys}. Details:"
+            f" {self.details}."
+        )
+    elif any(
         key not in self.details
         for key in (
             constants.R_SQUARED,
@@ -364,6 +377,24 @@ class GoodnessOfFitCheckResult(CheckResult):
           f" {self.details}."
       )
 
+  @property
+  def recommendation(self) -> str:
+    """Returns the check result message."""
+    if self.is_holdout:
+      report_str = (
+          "R-squared = {r_squared_all:.4f} (All),"
+          " {r_squared_train:.4f} (Train), {r_squared_test:.4f} (Test); MAPE"
+          " = {mape_all:.4f} (All), {mape_train:.4f} (Train),"
+          " {mape_test:.4f} (Test); wMAPE = {wmape_all:.4f} (All),"
+          " {wmape_train:.4f} (Train), {wmape_test:.4f} (Test)".format(
+              **self.details
+          )
+      )
+    else:
+      report_str = self.case.message_template.format(**self.details)
+
+    return f"{report_str}. {self.case.recommendation}"
+
 
 # ==============================================================================
 # Check: ROI Consistency

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/summarizer.py

@@ -21,11 +21,11 @@ import os
 import jinja2
 from meridian import constants as c
 from meridian.analysis import analyzer
-from meridian.analysis import formatter
 from meridian.analysis import summary_text
 from meridian.analysis import visualizer
 from meridian.data import time_coordinates as tc
 from meridian.model import model
+from meridian.templates import formatter
 import pandas as pd
 import xarray as xr
 

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/analysis/visualizer.py

@@ -22,9 +22,9 @@ import altair as alt
 from meridian import backend
 from meridian import constants as c
 from meridian.analysis import analyzer
-from meridian.analysis import formatter
 from meridian.analysis import summary_text
 from meridian.model import model
+from meridian.templates import formatter
 import numpy as np
 import pandas as pd
 import xarray as xr

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/backend/__init__.py

@@ -609,7 +609,37 @@ def _tf_get_seed_data(seed: Any) -> Optional[np.ndarray]:
 
 
 def _jax_convert_to_tensor(data, dtype=None):
-  """Converts data to a JAX array, handling strings as NumPy arrays."""
+  """Converts data to a JAX array, handling strings as NumPy arrays.
+
+  This function explicitly unwraps objects with a `.values` attribute (e.g.,
+  pandas.DataFrame, xarray.DataArray) to access the underlying NumPy array,
+  provided that `.values` is not a method. This takes precedence over the
+  `__array__` protocol.
+
+  It also handles precision mismatches: if `data` is float64 and `dtype` is
+  not specified, and JAX x64 mode is disabled (default), it issues a warning
+  and explicitly casts to float32 to match the backend default and prevent
+  silent precision loss or type errors in downstream operations.
+
+  Args:
+    data: The data to convert.
+    dtype: The desired data type.
+
+  Returns:
+    A JAX array, or a NumPy array if the dtype is a string type.
+  """
+  # Unwrap xarray.DataArray, pandas.Series, and pandas.DataFrame objects.
+  # These objects wrap the underlying NumPy array in a .values attribute.
+  if hasattr(data, "values") and not callable(data.values):
+    data = data.values
+
+  # Convert to numpy array upfront to simplify dtype inspection below.
+  # A standard Python float is 64-bit, and this conversion allows the
+  # subsequent logic to correctly detect and handle potential float64
+  # downcasting for scalar inputs.
+  if isinstance(data, (list, tuple, float)):
+    data = np.array(data)
+
   # JAX does not natively support string tensors in the same way TF does.
   # If a string dtype is requested, or if the data is inherently strings,
   # we fall back to a standard NumPy array.
@@ -623,13 +653,31 @@ def _jax_convert_to_tensor(data, dtype=None):
       # let jax.asarray handle it.
       pass
 
-  is_string_data = isinstance(data, (list, np.ndarray)) and np.array(
-      data
-  ).dtype.kind in ("S", "U")
+  is_string_data = isinstance(data, np.ndarray) and data.dtype.kind in (
+      "S",
+      "U",
+  )
 
-  if is_string_target or (dtype is None and is_string_data):
+  if is_string_target:
     return np.array(data, dtype=dtype)
 
+  if dtype is None and is_string_data:
+    return data
+
+  # If the user provides float64 data but does not request a specific dtype,
+  # and JAX 64-bit mode is disabled (default), JAX would implicitly truncate.
+  # We cast to float32 and warn the user to prevent silent mismatches.
+  if dtype is None:
+    is_float64_input = hasattr(data, "dtype") and data.dtype == np.float64
+    if is_float64_input:
+      if not jax.config.jax_enable_x64:
+        warnings.warn(
+            "Input data is float64. Casting to float32 to match backend "
+            "default precision.",
+            UserWarning,
+        )
+        dtype = jax_ops.float32
+
   return jax_ops.asarray(data, dtype=dtype)
 
 

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/backend/test_utils.py

@@ -14,6 +14,7 @@
 
 """Common testing utilities for Meridian, designed to be backend-agnostic."""
 
+import dataclasses
 from typing import Any, Optional
 
 from absl.testing import parameterized
@@ -26,6 +27,8 @@ import numpy as np
 from tensorflow.python.util.protobuf import compare
 # pylint: disable=g-direct-tensorflow-import
 from tensorflow.core.framework import tensor_pb2
+
+
 # pylint: enable=g-direct-tensorflow-import
 
 FieldDescriptor = descriptor.FieldDescriptor
@@ -80,6 +83,75 @@ def assert_allequal(a: ArrayLike, b: ArrayLike, err_msg: str = ""):
   np.testing.assert_array_equal(np.array(a), np.array(b), err_msg=err_msg)
 
 
+def assert_deep_equals(
+    test_case,
+    obj1: Any,
+    obj2: Any,
+    msg: str = "",
+    rtol: float = 1e-5,
+    atol: float = 1e-5,
+):
+  """Recursive equality check handling Dataclasses, Lists, and Backend Tensors.
+
+  Args:
+    test_case: The unittest.TestCase instance (self) to use for assertions.
+    obj1: The first object to compare.
+    obj2: The second object to compare.
+    msg: Optional error message prefix.
+    rtol: Relative tolerance for float comparison.
+    atol: Absolute tolerance for float comparison.
+  """
+  if obj1 is None or obj2 is None:
+    test_case.assertEqual(obj1, obj2, msg=msg)
+    return
+
+  if (
+      hasattr(obj1, "__array__")
+      or hasattr(obj1, "numpy")
+      or isinstance(obj1, (np.ndarray, backend.Tensor))
+  ):
+    arr1 = np.array(obj1)
+    arr2 = np.array(obj2)
+
+    # Check for non-numeric types where atol/rtol don't apply
+    if arr1.dtype.kind in ("U", "S", "O", "b"):
+      np.testing.assert_array_equal(arr1, arr2, err_msg=msg)
+    else:
+      np.testing.assert_allclose(arr1, arr2, err_msg=msg, rtol=rtol, atol=atol)
+    return
+
+  if dataclasses.is_dataclass(obj1):
+    test_case.assertIs(
+        type(obj1),
+        type(obj2),
+        msg=f"{msg} Type mismatch: {type(obj1)} vs {type(obj2)}",
+    )
+    for field in dataclasses.fields(obj1):
+      val1 = getattr(obj1, field.name)
+      val2 = getattr(obj2, field.name)
+      assert_deep_equals(
+          test_case,
+          val1,
+          val2,
+          msg=f"{msg}.{field.name}",
+          rtol=rtol,
+          atol=atol,
+      )
+    return
+
+  if isinstance(obj1, (list, tuple)):
+    test_case.assertIsInstance(obj2, (list, tuple), msg=f"{msg} Type mismatch")
+    test_case.assertEqual(len(obj1), len(obj2), msg=f"{msg} Length mismatch")
+    for i, (item1, item2) in enumerate(zip(obj1, obj2)):
+      assert_deep_equals(
+          test_case, item1, item2, msg=f"{msg}[{i}]", rtol=rtol, atol=atol
+      )
+    return
+
+  # Fallback to standard equality for primitives (int, str, float, etc.)
+  test_case.assertEqual(obj1, obj2, msg=msg)
+
+
 def assert_seed_allequal(a: Any, b: Any, err_msg: str = ""):
   """Backend-agnostic assertion to check if two seed objects are equal."""
   data_a = backend.get_seed_data(a)

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/constants.py

@@ -662,6 +662,7 @@ CURRENT_SPEND = 'current_spend'
 
 # Media summary metrics.
 SPEND = 'spend'
+COST = 'cost'
 IMPRESSIONS = 'impressions'
 ROI = 'roi'
 OPTIMIZED_ROI = 'optimized_roi'

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/data/load.py

@@ -35,6 +35,8 @@ __all__ = [
     'InputDataLoader',
     'XrDatasetDataLoader',
     'DataFrameDataLoader',
+    'CoordToColumns',
+    'CsvDataLoader',
 ]
 
 

{google_meridian-1.3.1 → google_meridian-1.3.2}/meridian/model/eda/__init__.py

@@ -17,4 +17,3 @@
 from meridian.model.eda import eda_engine
 from meridian.model.eda import eda_outcome
 from meridian.model.eda import eda_spec
-from meridian.model.eda import meridian_eda