use dictionary for coords

williambdean · ricardoV94 · commit f69e159cb834 · 2025-11-21T16:43:27.000+01:00
diff --git a/pymc/model/core.py b/pymc/model/core.py
@@ -368,10 +368,8 @@ class Model(WithMemoization, metaclass=ContextMeta):
         import numpy as np
 
         coords = {
-            "feature",
-            ["A", "B", "C"],
-            "trial",
-            [1, 2, 3, 4, 5],
+            "feature": ["A", "B", "C"],
+            "trial": [1, 2, 3, 4, 5],
         }
 
         with pm.Model(coords=coords) as model:
