Set accepts sparse tag to True (#893)

Quantco · Jan 3, 2025 · 5a4325d · 5a4325d
1 parent 84fdec6
commit 5a4325d
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/glum/_glm.py b/src/glum/_glm.py
@@ -798,6 +798,11 @@ def __init__(
         self.cat_missing_method = cat_missing_method
         self.cat_missing_name = cat_missing_name
 
+    def __sklearn_tags__(self):
+        tags = super().__sklearn_tags__()
+        tags.input_tags.sparse = True
+        return tags
+
     @property
     def family_instance(self) -> ExponentialDispersionModel:
         """Return an :class:`~glum._distribution.ExponentialDispersionModel`."""