lint

Author: Samoed

autointent/_pipeline/_pipeline.py

@@ -10,8 +10,6 @@
 
 from autointent import Context, Dataset
 from autointent.configs import InferenceNodeConfig, LoggingConfig, VectorIndexConfig
-from autointent.custom_types import ListOfGenericLabels, NodeType
-from autointent.configs import CrossEncoderConfig, EmbedderConfig, InferenceNodeConfig, LoggingConfig, VectorIndexConfig
 from autointent.custom_types import ListOfGenericLabels, NodeType, ValidationScheme
 from autointent.metrics import PREDICTION_METRICS_MULTILABEL
 from autointent.nodes import InferenceNode, NodeOptimizer

autointent/modules/embedding/_logreg.py

@@ -1,6 +1,6 @@
 """LogregAimedEmbedding class for a proxy optimzation of embedding."""
 
-from typing import Any, Literal
+from typing import Any
 
 import numpy as np
 from numpy.typing import NDArray

autointent/modules/embedding/_retrieval.py

@@ -1,6 +1,6 @@
 """RetrievalAimedEmbedding class for a proxy optimization of embedding."""
 
-from typing import Any, Literal
+from typing import Any
 
 from autointent import Context, VectorIndex
 from autointent.context.optimization_info import RetrieverArtifact