Skip to content

Commit d9ea833

Browse files
committed
Fix GLasso import for SDML for newer sklearn versions
1 parent 8fb6872 commit d9ea833

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

metric_learn/sdml.py

+5-1
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,11 @@
66
import numpy as np
77
from sklearn.base import TransformerMixin
88
from scipy.linalg import pinvh
9-
from sklearn.covariance import graphical_lasso
9+
try:
10+
from sklearn.covariance import _graphical_lasso as graphical_lasso
11+
except ImportError:
12+
from sklearn.covariance import graphical_lasso
13+
1014
from sklearn.exceptions import ConvergenceWarning
1115

1216
from .base_metric import MahalanobisMixin, _PairsClassifierMixin

0 commit comments

Comments
 (0)