From bf74e01466cc33d08152fd97adae33e6dc8b878e Mon Sep 17 00:00:00 2001 From: Daniel Nouri Date: Tue, 6 Nov 2018 13:58:57 +0100 Subject: [PATCH] Do not run OpenML tests; buggy with scikit-learn<0.20.1 --- palladium/dataset.py | 2 +- palladium/tests/test_dataset.py | 3 +++ 2 files changed, 4 insertions(+), 1 deletion(-) diff --git a/palladium/dataset.py b/palladium/dataset.py index 0723a26..cd6f462 100644 --- a/palladium/dataset.py +++ b/palladium/dataset.py @@ -108,7 +108,7 @@ def __call__(self): return data, target -class OpenML(DatasetLoader): +class OpenML(DatasetLoader): # pragma: no cover """A :class:`~palladium.interfaces.DatasetLoader` that uses scikit-learn's :func:`sklearn.datasets.fetch_openml` to load data from OpenML. diff --git a/palladium/tests/test_dataset.py b/palladium/tests/test_dataset.py index 6ce1dde..37649a4 100644 --- a/palladium/tests/test_dataset.py +++ b/palladium/tests/test_dataset.py @@ -6,6 +6,7 @@ import numpy as np from pandas import DataFrame import pytest +import sklearn dummy_dataframe = DataFrame({ 'datacol1': [10, 11, 12, 13, 14], @@ -135,6 +136,8 @@ def test_concurrency(self, sql): [th.join() for th in threads] +@pytest.mark.skipif(sklearn.__version__ < "0.20.1", + reason="scikit-learn version too old") class TestOpenML: @pytest.fixture def OpenML(self):