Skip to content

Commit 6431135

Browse files
committed
FIX? container building
do not import openml inside a container while running the tests as it can break because of certain directories not existing.
1 parent a4c0937 commit 6431135

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

test/test_pipeline/implementations/test_SparseOneHotEncoder.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
import numpy as np
44

55
import scipy.sparse
6-
import openml
76
import sklearn.tree
87
import sklearn.datasets
98
import sklearn.model_selection
@@ -91,8 +90,8 @@ def test_transform_with_unknown_value(self):
9190
self.assertEqual(3, np.sum(output))
9291

9392
def test_classification_workflow(self):
94-
task = openml.tasks.get_task(254)
95-
X, y = task.get_X_and_y()
93+
X, y = sklearn.datasets.fetch_openml(data_id=24, as_frame=False, return_X_y=True)
94+
print(type(X))
9695

9796
X_train, X_test, y_train, y_test = \
9897
sklearn.model_selection.train_test_split(X, y, random_state=3,

0 commit comments

Comments
 (0)