diff --git a/bambi/backend/terms.py b/bambi/backend/terms.py index 45439d9ad..7c3086b02 100644 --- 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 index 20a4b4887..9207d2908 100644 --- 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 index 92d70f43e..775d83c4b 100644 --- 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):