fix import config

Author: Samoed

autointent/modules/abc/_base.py

@@ -14,7 +14,6 @@
 from autointent.context.optimization_info import Artifact
 from autointent.custom_types import ListOfGenericLabels, ListOfLabels
 from autointent.exceptions import WrongClassificationError
-from autointent.schemas import EmbedderConfig
 
 logger = logging.getLogger(__name__)
 
@@ -115,7 +114,7 @@ def from_context(cls, context: Context, **kwargs: dict[str, Any]) -> "Module":
         :param kwargs: Additional kwargs.
         """
 
-    def get_embedder_config(self) -> EmbedderConfig | None:
+    def get_embedder_config(self) -> dict[str, Any] | None:
         """
         Get the config of the embedder.
 

autointent/modules/scoring/_description/description.py

@@ -72,13 +72,13 @@ def from_context(
             embedder_config=embedder_config,
         )
 
-    def get_embedder_config(self) -> EmbedderConfig:
+    def get_embedder_config(self) -> dict[str, Any]:
         """
         Get the name of the embedder.
 
         :return: Embedder name.
         """
-        return self.embedder_config
+        return self.embedder_config.model_dump()
 
     def fit(
         self,

autointent/modules/scoring/_knn/knn.py

@@ -101,13 +101,13 @@ def from_context(
             weights=weights,
         )
 
-    def get_embedder_config(self) -> EmbedderConfig:
+    def get_embedder_config(self) -> dict[str, Any]:
         """
         Get the name of the embedder.
 
         :return: Embedder name.
         """
-        return self.embedder_config
+        return self.embedder_config.model_dump()
 
     def fit(self, utterances: list[str], labels: ListOfLabels, clear_cache: bool = False) -> None:
         """

autointent/modules/scoring/_linear.py

@@ -91,13 +91,13 @@ def from_context(
             embedder_config=embedder_config,
         )
 
-    def get_embedder_config(self) -> EmbedderConfig:
+    def get_embedder_config(self) -> dict[str, Any]:
         """
         Get the name of the embedder.
 
         :return: Embedder name.
         """
-        return self.embedder_config
+        return self.embedder_config.model_dump()
 
     def fit(
         self,

autointent/modules/scoring/_mlknn/mlknn.py

@@ -105,13 +105,13 @@ def from_context(
             ignore_first_neighbours=ignore_first_neighbours,
         )
 
-    def get_embedder_config(self) -> EmbedderConfig:
+    def get_embedder_config(self) -> dict[str, Any]:
         """
         Get the name of the embedder.
 
         :return: Embedder name.
         """
-        return self.embedder_config
+        return self.embedder_config.model_dump()
 
     def fit(self, utterances: list[str], labels: ListOfLabels) -> None:
         """