use fit_posterior

pymc-labs · Apr 18, 2024 · 89ca035 · 89ca035
1 parent f9dc0dc
commit 89ca035
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/pymc_marketing/clv/models/gamma_gamma.py b/pymc_marketing/clv/models/gamma_gamma.py
@@ -56,10 +56,9 @@ def expected_customer_spend(
         mean_transaction_value, frequency = to_xarray(
             customer_id, mean_transaction_value, frequency
         )
-        if self.idata is None:
+        if (posterior := self.fit_result) is None:
             raise ValueError("Model must be fitted first")
 
-        posterior = self.idata.posterior
         p = posterior["p"]
         q = posterior["q"]
         v = posterior["v"]
@@ -92,10 +91,9 @@ def distribution_new_customer_spend(
     def expected_new_customer_spend(self) -> xarray.DataArray:
         """Expected transaction value for a new customer"""
 
-        if self.idata is None:
+        if (posterior := self.fit_result) is None:
             raise ValueError("Model must be fitted first")
 
-        posterior = self.idata.post
         p_mean = posterior["p"]
         q_mean = posterior["q"]
         v_mean = posterior["v"]