add compatibility with older np versions

csinva · Aug 16, 2024 · 7581f43 · 7581f43
1 parent c9bb163
commit 7581f43
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/imodels/util/score.py b/imodels/util/score.py
@@ -3,7 +3,10 @@
 
 import pandas as pd
 import numpy as np
-from sklearn.utils._mask import indices_to_mask
+try:
+    from sklearn.utils._mask import indices_to_mask
+except:
+    from sklearn.utils import indices_to_mask
 from sklearn.linear_model import Lasso, LogisticRegression
 from sklearn.linear_model._coordinate_descent import _alpha_grid
 from sklearn.model_selection import cross_val_score