bumping up the version of sklearn daal4py patches were tested with
@@ -84,10 +84,13 @@ def do_unpatch(name):
def enable(name=None):
if LooseVersion(sklearn_version) < LooseVersion("0.20.0"):
raise NotImplementedError("daal4py patches apply for scikit-learn >= 0.20.0 only ...")
- elif LooseVersion(sklearn_version) > LooseVersion("0.21.1"):
+ elif LooseVersion(sklearn_version) > LooseVersion("0.21.2"):
warn_msg = ("daal4py {daal4py_version} has only been tested " +
- "with scikit-learn 0.21.1, found version: {sklearn_version}")
- warnings.warn(warn_msg.format(daal4py_version=daal4py_version, sklearn_version=sklearn_version))
+ "with scikit-learn 0.21.2, found version: {sklearn_version}")
+ warnings.warn(warn_msg.format(
+ daal4py_version=daal4py_version,
+ sklearn_version=sklearn_version)
+ )
if name is not None:
do_patch(name)
