Rename control coordinate

pymc-labs · Feb 22, 2023 · d410c11 · d410c11
1 parent bd60258
commit d410c11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc_marketing/mmm/delayed_saturated_mmm.py b/pymc_marketing/mmm/delayed_saturated_mmm.py
@@ -57,7 +57,7 @@ def build_model(
         }
 
         if control_data is not None:
-            coords["control_names"] = control_data.columns
+            coords["control"] = control_data.columns
 
         with pm.Model(coords=coords) as self.model:
             channel_data_ = pm.MutableData(