bambinos · tomicapretto · Oct 10, 2023 · Oct 6, 2023 · Oct 6, 2023 · Oct 6, 2023
diff --git a/bambi/interpret/utils.py b/bambi/interpret/utils.py
@@ -264,7 +264,12 @@ def enforce_dtypes(data: pd.DataFrame, df: pd.DataFrame, except_col=None) -> pd.
     observed_dtypes = data.dtypes
     for col in df.columns:
         if col in observed_dtypes.index and not except_col:
-            df[col] = df[col].astype(observed_dtypes[col])
+            if observed_dtypes[col] == "category":
+                # explicitly converts to category dtype
+                df[col] = df[col].astype("category")
+            else:
+                # casts the original dtype to the new data
+                df[col] = df[col].astype(observed_dtypes[col])
     return df
 
 

diff --git a/docs/_quarto.yml b/docs/_quarto.yml
@@ -52,6 +52,7 @@ website:
             - notebooks/Strack_RRR_re_analysis.ipynb
             - notebooks/shooter_crossed_random_ANOVA.ipynb
             - notebooks/t_regression.ipynb
+            - notebooks/predict_new_groups.ipynb
         - section: Generalized linear models
           contents:
               - notebooks/logistic_regression.ipynb

diff --git a/docs/notebooks/gallery.yml b/docs/notebooks/gallery.yml
@@ -37,6 +37,10 @@
       subtitle: With the Student's T distribution
       href: t_regression.ipynb
       thumbnail: thumbnails/t_regression.png
+    - title: Predict new groups
+      subtitle: With Hierarchical models
+      href: predict_new_groups.ipynb
+      thumbnail: thumbnails/predict_new_group.png
 - category: Generalized Linear Models
   description: ""
   tiles:

diff --git a/docs/notebooks/predict_new_groups.ipynb b/docs/notebooks/predict_new_groups.ipynb
diff --git a/docs/notebooks/thumbnails/predict_new_group.png b/docs/notebooks/thumbnails/predict_new_group.png