Skip to content

Commit df70dad

Browse files
committed
Fix import errors introduced by sklearn 0.1.5
With sklearn version 0.1.5 the check_pandas_support function has been moved to util._optional_dependencies. This has been adapted in our code.
1 parent dd31493 commit df70dad

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/skmatter/datasets/_base.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,15 @@
11
from os.path import dirname, join
22

33
import numpy as np
4-
from sklearn.utils import Bunch, check_pandas_support
4+
5+
import skmatter
6+
7+
8+
if skmatter.__version__ >= "0.1.5":
9+
from skmatter.utils._optional_dependencies import check_pandas_support
10+
else:
11+
from sklearn.utils import check_pandas_support
12+
from sklearn.utils import Bunch
513

614

715
def load_nice_dataset():

0 commit comments

Comments
 (0)