Skip to content

Commit 5d0ba0b

Browse files
authored
Merge pull request #7 from skorch-dev/master
Move away from .scoring imports (skorch-dev#575)
2 parents 5546553 + b7606bd commit 5d0ba0b

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

skorch/callbacks/scoring.py

+8-2
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,18 @@
22

33
from contextlib import contextmanager
44
from contextlib import suppress
5+
from distutils.version import LooseVersion
56
from functools import partial
67
import warnings
78

89
import numpy as np
9-
from sklearn.metrics.scorer import (
10-
check_scoring, _BaseScorer, make_scorer)
10+
import sklearn
11+
from sklearn.metrics import make_scorer, check_scoring
12+
13+
if LooseVersion(sklearn.__version__) >= '0.22':
14+
from sklearn.metrics._scorer import _BaseScorer
15+
else:
16+
from sklearn.metrics.scorer import _BaseScorer
1117

1218
from skorch.utils import data_from_dataset
1319
from skorch.utils import is_skorch_dataset

0 commit comments

Comments
 (0)