Refactor and enhance SensitivityAnalysis for MMM

pymc_marketing/mmm/multidimensional.py

@@ -1712,7 +1712,10 @@ def sensitivity(self) -> SensitivityAnalysis:
         ...     sweep_type="multiplicative",
         ... )
         """
-        return SensitivityAnalysis(mmm=self)
+        # Provide the underlying PyMC model, the model's inference data, and dims
+        return SensitivityAnalysis(
+            pymc_model=self.model, idata=self.idata, dims=self.dims
+        )
 
     def _make_channel_transform(
         self, df_lift_test: pd.DataFrame