Prevent underscores from being removed in dim names (#664)

* add underscores to dim names * Fix dim names multinomial family * Fix dimname categorical family
bambinos · Apr 7, 2023 · d9ca5d9 · d9ca5d9
1 parent 341966f
commit d9ca5d9
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/bambi/backend/terms.py b/bambi/backend/terms.py
@@ -126,7 +126,7 @@ def get_coords(self):
         new_coords = {}
         for key, value in coords.items():
             _, kind = key.split("__")
-            new_coords[self.term.alias + kind] = value
+            new_coords[self.term.alias + "__" + kind] = value
         return new_coords
 
     def build_distribution(self, prior, label, **kwargs):

diff --git a/bambi/families/multivariate.py b/bambi/families/multivariate.py
@@ -73,7 +73,7 @@ def posterior_predictive(self, model, posterior, **kwargs):
 
     def get_coords(self, response):
         # For the moment, it always uses the first column as reference.
-        name = response.name + "_dim"
+        name = get_aliased_name(response) + "_dim"
         labels = self.get_levels(response)
         return {name: labels[1:]}
 

diff --git a/bambi/families/univariate.py b/bambi/families/univariate.py
@@ -93,7 +93,7 @@ def get_data(self, response):
         return np.nonzero(response.term.data)[1]
 
     def get_coords(self, response):
-        name = response.name + "_dim"
+        name = get_aliased_name(response) + "_dim"
         return {name: [level for level in response.levels if level != response.reference]}
 
     def get_reference(self, response):