Add private module aliases for scipy.sparse

Signed-off-by: Connor Tann <[email protected]>
interpretml · Mar 7, 2024 · 54fe015 · 54fe015
1 parent 12964ec
commit 54fe015
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/slicer/slicer_internal.py b/slicer/slicer_internal.py
@@ -567,6 +567,8 @@ class UnifiedDataHandler:
         ("numpy", "ndarray"): ArrayHandler,
         ("scipy.sparse.csc", "csc_matrix"): ArrayHandler,
         ("scipy.sparse.csr", "csr_matrix"): ArrayHandler,
+        ("scipy.sparse._csc", "csc_matrix"): ArrayHandler,
+        ("scipy.sparse._csr", "csr_matrix"): ArrayHandler,
         ("scipy.sparse.dok", "dok_matrix"): ArrayHandler,
         ("scipy.sparse.lil", "lil_matrix"): ArrayHandler,
         ("pandas.core.frame", "DataFrame"): DataFrameHandler,