convert

Quantco · Jan 30, 2024 · 8504c03 · 8504c03
1 parent f87a772
commit 8504c03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/glum/_glm.py b/src/glum/_glm.py
@@ -896,7 +896,7 @@ def _convert_from_pandas(
                 ),
                 getattr(self, "cat_missing_method", "fail"),
             )
-            if getattr(self, "cat_missing_method", "fail"):
+            if getattr(self, "cat_missing_method", "fail") == "convert":
                 df = _add_missing_categories(
                     df=df,
                     dtypes=self.feature_dtypes_,