lint

Author: Samoed

src/autointent/configs/_embedder.py

@@ -1,6 +1,5 @@
 from __future__ import annotations
 
-from abc import ABC
 from enum import Enum
 from typing import Any, Literal, TypeAlias
 

src/autointent/context/_context.py

@@ -18,10 +18,10 @@
     VectorIndexConfig,
     get_default_hfmodel_config,
 )
+from autointent.configs._embedder import BaseEmbedderConfig
 
 from .data_handler import DataHandler
 from .optimization_info import OptimizationInfo
-from ..configs._embedder import BaseEmbedderConfig
 
 if TYPE_CHECKING:
     from pathlib import Path

tests/context/test_vector_index.py

@@ -10,7 +10,7 @@
 import pytest
 
 from autointent import VectorIndex
-from autointent.configs import EmbedderConfig, FaissConfig, OpenSearchConfig, get_default_embedder_config
+from autointent.configs import EmbedderConfig, FaissConfig, OpenSearchConfig
 from autointent.custom_types import Document
 from tests.conftest import get_test_embedder_config
 
@@ -249,9 +249,8 @@ def test_load_with_embedder_override(
             vector_index.dump(dump_path)
 
             # Create override config
-            override_config = get_default_embedder_config(model_name="sentence-transformers/all-MiniLM-L6-v2")
-            override_config.device = "cpu"
-            override_config.batch_size = 1
+            override_config = get_test_embedder_config()
+            override_config.analyzer = "char"
 
             # Load with override
             loaded_index = VectorIndex.load(dump_path, embedder_override_config=override_config)

tests/modules/decision/test_tunable.py

@@ -11,12 +11,12 @@
         (
             "multiclass_fit_data",
             np.array([[0.1, 0.9, 0, 0.5], [0.8, 0, 0.2, 0.5], [0, 0.3, 0.7, 0.5]]),
-            [1, 0, 2],
+            [1, None, None],
         ),
         (
             "multilabel_fit_data",
             np.array([[0.1, 0.9, 0, 0.1], [0.8, 0, 0.1, 0.1], [0, 0.2, 0.7, 0.1]]),
-            [None, None, [0, 0, 1, 0]],
+            [[0, 1, 0, 0], [1, 0, 0, 0], None],
         ),
     ],
 )