remove unnecessary setup_environment() calls

Author: voorhs

tests/context/datahandler/test_multilabel_generation.py

@@ -4,7 +4,6 @@
 
 from autointent import VectorIndex
 from autointent.context.data_handler import DataHandler
-from tests.conftest import setup_environment
 
 
 @pytest.fixture
@@ -16,6 +15,5 @@ def mock_data_handler():
 
 
 def test_vector_index_initialization():
-    dump_dir, logs_dir = setup_environment()
     index = VectorIndex(embedder_device="cpu", embedder_model_name="sergeyzh/rubert-tiny-turbo")
     assert index.embedder_device == "cpu"

tests/context/test_vector_index.py

@@ -1,7 +1,6 @@
 import pytest
 
 from autointent import VectorIndex
-from tests.conftest import setup_environment
 
 
 @pytest.fixture
@@ -14,7 +13,6 @@ class MockDataHandler:
 
 
 def test_create_collection(data_handler):
-    dump_dir, logs_dir = setup_environment()
     vector_index = VectorIndex(embedder_model_name="bert-base-uncased", embedder_device="cpu")
     vector_index.add(
         data_handler.utterances_train,

tests/modules/decision/conftest.py

@@ -3,13 +3,10 @@
 
 from autointent.context.data_handler import DataHandler
 from autointent.modules import KNNScorer
-from tests.conftest import setup_environment
 
 
 @pytest.fixture
 def multiclass_fit_data(dataset):
-    dump_dir, logs_dir = setup_environment()
-
     data_handler = DataHandler(dataset)
 
     knn_params = {
@@ -27,8 +24,6 @@ def multiclass_fit_data(dataset):
 
 @pytest.fixture
 def multilabel_fit_data(dataset):
-    dump_dir, logs_dir = setup_environment()
-
     data_handler = DataHandler(dataset.to_multilabel())
 
     knn_params = {

tests/modules/retrieval/test_vectordb.py

@@ -1,9 +1,7 @@
 from autointent.modules.embedding import RetrievalEmbedding
-from tests.conftest import setup_environment
 
 
 def test_get_assets_returns_correct_artifact():
-    dump_dir, logs_dir = setup_environment()
     module = RetrievalEmbedding(k=5, embedder_name="sergeyzh/rubert-tiny-turbo")
     artifact = module.get_assets()
     assert artifact.embedder_name == "sergeyzh/rubert-tiny-turbo"

tests/modules/scoring/test_description.py

@@ -3,7 +3,6 @@
 
 from autointent.context.data_handler import DataHandler
 from autointent.modules import DescriptionScorer
-from tests.conftest import setup_environment
 
 
 @pytest.mark.parametrize(
@@ -14,7 +13,6 @@
     ],
 )
 def test_description_scorer(dataset, expected_prediction, multilabel):
-    dump_dir, logs_dir = setup_environment()
     if multilabel:
         dataset = dataset.to_multilabel()
     data_handler = DataHandler(dataset)

tests/modules/scoring/test_dnnc.py

@@ -3,13 +3,10 @@
 
 from autointent.context.data_handler import DataHandler
 from autointent.modules import DNNCScorer
-from tests.conftest import setup_environment
 
 
 @pytest.mark.parametrize(("train_head", "pred_score"), [(True, 1)])
 def test_base_dnnc(dataset, train_head, pred_score):
-    dump_dir, logs_dir = setup_environment()
-
     data_handler = DataHandler(dataset)
 
     scorer = DNNCScorer(

tests/modules/scoring/test_knn.py

@@ -2,12 +2,9 @@
 
 from autointent.context.data_handler import DataHandler
 from autointent.modules import KNNScorer
-from tests.conftest import setup_environment
 
 
 def test_base_knn(dataset):
-    dump_dir, logs_dir = setup_environment()
-
     data_handler = DataHandler(dataset)
 
     scorer = KNNScorer(k=3, weights="distance", embedder_name="sergeyzh/rubert-tiny-turbo", embedder_device="cpu")

tests/modules/scoring/test_linear.py

@@ -2,11 +2,9 @@
 
 from autointent.context.data_handler import DataHandler
 from autointent.modules import LinearScorer
-from tests.conftest import setup_environment
 
 
 def test_base_linear(dataset):
-    dump_dir, logs_dir = setup_environment()
 
     data_handler = DataHandler(dataset)
 

tests/modules/scoring/test_mlknn.py

@@ -4,12 +4,9 @@
 from autointent.context.data_handler import DataHandler
 from autointent.custom_types import Split
 from autointent.modules.scoring import MLKnnScorer
-from tests.conftest import setup_environment
 
 
 def test_base_mlknn(dataset):
-    dump_dir, logs_dir = setup_environment()
-
     dataset[Split.TEST] = HFDataset.from_list(
         [
             {

tests/modules/scoring/test_rerank_scorer.py

@@ -2,12 +2,9 @@
 
 from autointent.context.data_handler import DataHandler
 from autointent.modules import RerankScorer
-from tests.conftest import setup_environment
 
 
 def test_base_rerank_scorer(dataset):
-    dump_dir, logs_dir = setup_environment()
-
     data_handler = DataHandler(dataset)
 
     scorer = RerankScorer(