Skip to content

Commit 28f894b

Browse files
committed
fix sklearn test
1 parent 50b55e8 commit 28f894b

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/modules/scoring/test_sklearn.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,12 +2,9 @@
22

33
from autointent.context.data_handler import DataHandler
44
from autointent.modules import SklearnScorer
5-
from tests.conftest import setup_environment
65

76

8-
def test_base_linear(dataset):
9-
dump_dir, logs_dir = setup_environment()
10-
7+
def test_base_sklearn(dataset):
118
data_handler = DataHandler(dataset)
129

1310
scorer = SklearnScorer(embedder_name="sergeyzh/rubert-tiny-turbo", clf_name="LogisticRegression")

0 commit comments

Comments
 (0)