quick changes

Author: cetagostini

causalpy/pymc_models.py

@@ -1064,10 +1064,14 @@ def build_model(
 
             # Time data for trend and seasonality
             t_trend_data = pm.Data(
-                "t_trend_data", time_for_trend, dims="obs_ind", mutable=True
+                "t_trend_data",
+                time_for_trend,
+                dims="obs_ind",
             )
             t_season_data = pm.Data(
-                "t_season_data", time_for_seasonality, dims="obs_ind", mutable=True
+                "t_season_data",
+                time_for_seasonality,
+                dims="obs_ind",
             )
 
             # Get validated components (no more ugly imports in build_model!)
@@ -1114,8 +1118,8 @@ def build_model(
                         f"Shape mismatch: X_values_for_pymc has {X_values_for_pymc.shape[1]} columns, but "
                         f"{len(self._exog_var_names)} names in self._exog_var_names ({self._exog_var_names})."
                     )
-                X_data = pm.Data(
-                    "X", X_values_for_pymc, dims=["obs_ind", "coeffs"], mutable=True
+                X_data = pm.MutableData(
+                    "X", X_values_for_pymc, dims=["obs_ind", "coeffs"]
                 )
                 beta = pm.Normal("beta", mu=0, sigma=10, dims="coeffs")
                 mu_ = mu_ + pm.math.dot(X_data, beta)
@@ -1125,7 +1129,7 @@ def build_model(
 
             # Likelihood
             sigma = pm.HalfNormal("sigma", sigma=self.prior_sigma)
-            y_data = pm.Data("y", y.flatten(), dims="obs_ind", mutable=True)
+            y_data = pm.MutableData("y", y.flatten(), dims="obs_ind")
             pm.Normal("y_hat", mu=mu, sigma=sigma, observed=y_data, dims="obs_ind")
 
     def fit(