Rename meridian_max_lag to max_lag in CalibrationBuilder.

PiperOrigin-RevId: 956088344

Author: The Meridian Authors

meridian/constants.py

@@ -233,6 +233,7 @@
 HILL_BEFORE_ADSTOCK = 'hill_before_adstock'
 ADSTOCK_MEMORY_OPTIMIZED = 'adstock_memory_optimized'
 MAX_LAG = 'max_lag'
+DEFAULT_MAX_LAG = 8
 UNIQUE_SIGMA_FOR_EACH_GEO = 'unique_sigma_for_each_geo'
 ROI_CALIBRATION_PERIOD = 'roi_calibration_period'
 RF_ROI_CALIBRATION_PERIOD = 'rf_roi_calibration_period'

meridian/model/prior_distribution.py

@@ -1251,7 +1251,9 @@ def distributions_are_equal(
     if isinstance(
         a_params[key], (backend.Tensor, np.ndarray, float, int)
     ) and isinstance(b_params[key], (backend.Tensor, np.ndarray, float, int)):
-      if not backend.allclose(a_params[key], b_params[key]):
+      a_val = backend.cast(a_params[key], backend.float_dtype)
+      b_val = backend.cast(b_params[key], backend.float_dtype)
+      if not backend.allclose(a_val, b_val):
         return False
     else:
       if np.any(a_params[key] != b_params[key]):

meridian/model/spec.py

@@ -234,7 +234,7 @@ class ModelSpec:
   )
   media_effects_dist: str = constants.MEDIA_EFFECTS_LOG_NORMAL
   hill_before_adstock: bool = False
-  max_lag: int = 8
+  max_lag: int = constants.DEFAULT_MAX_LAG
   unique_sigma_for_each_geo: bool = False
   media_prior_type: str | None = None
   rf_prior_type: str | None = None