diff options
author | Vincent Rouvreau <10407034+VincentRouvreau@users.noreply.github.com> | 2022-11-21 16:40:42 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-11-21 16:40:42 +0100 |
commit | a1957db266900049d4da7603d34edbc82cb38c9f (patch) | |
tree | ed48d18a77310cc4a0a7c1c5df6fce11e4afd399 | |
parent | 79a713843d79bc99deecd796e9ccfd2b5a55da01 (diff) | |
parent | 78533374f5255a601e4e12edf28ca8b51b93f016 (diff) |
Merge pull request #744 from mglisse/distancemetric
sklearn's DistanceMetric has moved
-rw-r--r-- | src/python/gudhi/representations/vector_methods.py | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/python/gudhi/representations/vector_methods.py b/src/python/gudhi/representations/vector_methods.py index a169aee8..d52185ef 100644 --- a/src/python/gudhi/representations/vector_methods.py +++ b/src/python/gudhi/representations/vector_methods.py @@ -13,8 +13,13 @@ import numpy as np from sklearn.base import BaseEstimator, TransformerMixin from sklearn.exceptions import NotFittedError from sklearn.preprocessing import MinMaxScaler, MaxAbsScaler -from sklearn.neighbors import DistanceMetric from sklearn.metrics import pairwise +try: + # New location since 1.0 + from sklearn.metrics import DistanceMetric +except ImportError: + # Will be removed in 1.3 + from sklearn.neighbors import DistanceMetric from .preprocessing import DiagramScaler, BirthPersistenceTransform |