From 7581f435e7e9d991643cbdef850e4e31e69ebdb7 Mon Sep 17 00:00:00 2001 From: Chandan Singh Date: Fri, 16 Aug 2024 00:37:10 -0700 Subject: [PATCH] add compatibility with older np versions --- imodels/util/score.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/imodels/util/score.py b/imodels/util/score.py index 4e94100d..982b4981 100644 --- 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