Merge bf74e01 into 76b2b19

ottogroup · Nov 6, 2018 · 145795d · 145795d
2 parents 76b2b19 + bf74e01
commit 145795d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git 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
@@ -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):