fix more docs

Author: Samoed

autointent/modules/embedding/_logreg.py

@@ -35,7 +35,7 @@ class LogregAimedEmbedding(EmbeddingModule):
         utterances = ["bye", "how are you?", "good morning"]
         labels = [0, 1, 1]
         retrieval = LogregAimedEmbedding(
-            embedder_name="sergeyzh/rubert-tiny-turbo",
+            embedder_config="sergeyzh/rubert-tiny-turbo",
             cv=2
         )
         retrieval.fit(utterances, labels)

autointent/modules/embedding/_retrieval.py

@@ -30,7 +30,7 @@ class RetrievalAimedEmbedding(EmbeddingModule):
         labels = [0, 1, 1]
         retrieval = RetrievalAimedEmbedding(
             k=2,
-            embedder_name="sergeyzh/rubert-tiny-turbo",
+            embedder_config="sergeyzh/rubert-tiny-turbo",
         )
         retrieval.fit(utterances, labels)
 

autointent/modules/scoring/_dnnc/dnnc.py

@@ -50,7 +50,7 @@ class DNNCScorer(ScoringModule):
         labels = [0, 1]
         scorer = DNNCScorer(
             cross_encoder_name="cross-encoder/ms-marco-MiniLM-L-6-v2",
-            embedder_name="sergeyzh/rubert-tiny-turbo",
+            embedder_config="sergeyzh/rubert-tiny-turbo",
             k=5,
         )
         scorer.fit(utterances, labels)

autointent/modules/scoring/_knn/knn.py

@@ -33,7 +33,7 @@ class KNNScorer(ScoringModule):
         utterances = ["hello", "how are you?"]
         labels = [0, 1]
         scorer = KNNScorer(
-            embedder_name="sergeyzh/rubert-tiny-turbo",
+            embedder_config="sergeyzh/rubert-tiny-turbo",
             k=5,
         )
         scorer.fit(utterances, labels)

autointent/modules/scoring/_linear.py

@@ -28,7 +28,7 @@ class LinearScorer(ScoringModule):
 
         from autointent.modules import LinearScorer
         scorer = LinearScorer(
-            embedder_name="sergeyzh/rubert-tiny-turbo", cv=2
+            embedder_config="sergeyzh/rubert-tiny-turbo", cv=2
         )
         utterances = ["hello", "goodbye", "allo", "sayonara"]
         labels = [0, 1, 0, 1]

autointent/modules/scoring/_mlknn/mlknn.py

@@ -30,7 +30,7 @@ class MLKnnScorer(ScoringModule):
         labels = [[1,0], [0,1]]
         scorer = MLKnnScorer(
             k=5,
-            embedder_name="sergeyzh/rubert-tiny-turbo",
+            embedder_config="sergeyzh/rubert-tiny-turbo",
         )
         scorer.fit(utterances, labels)
         test_utterances = ["Hi!", "What's up?"]

autointent/nodes/_optimization/_node_optimizer.py

@@ -64,9 +64,9 @@ def fit(self, context: Context) -> None:
                 )
                 module = self.node_info.modules_available[module_name].from_context(context, **module_kwargs)
 
-                embedder_name = module.get_embedder_name()
-                if embedder_name is not None:
-                    module_kwargs["embedder_config"] = embedder_name
+                embedder_config = module.get_embedder_name()
+                if embedder_config is not None:
+                    module_kwargs["embedder_config"] = embedder_config
 
                 self._logger.debug("optimizing %s module...", module_name)
                 self.module_fit(module, context)

tests/configs/test_embedding.py

@@ -40,7 +40,7 @@ def test_invalid_embedding_config_missing_field():
             "node_type": "embedding",
             # Missing "target_metric"
             "search_space": [
-                {"module_name": "retrieval", "embedder_name": ["sentence-transformers/all-MiniLM-L6-v2"], "k": [5, 10]}
+                {"module_name": "retrieval", "embedder_config": ["sentence-transformers/all-MiniLM-L6-v2"], "k": [5, 10]}
             ],
         }
     ]
@@ -58,7 +58,7 @@ def test_invalid_embedding_config_wrong_type():
             "search_space": [
                 {
                     "module_name": "logreg_embedding",
-                    "embedder_name": "not_a_list",  # Should be a list of strings
+                    "embedder_config": "not_a_list",  # Should be a list of strings
                     "cv": ["wrong_type"],  # Should be a list of integers
                 }
             ],

tests/configs/test_scoring.py

@@ -88,7 +88,7 @@ def test_invalid_scoring_config_wrong_type():
         "search_space": [
             {
                 "module_name": "knn",
-                "embedder_name": "should_be_list",  # Should be a list of strings
+                "embedder_config": "should_be_list",  # Should be a list of strings
                 "k": "not_an_int_list",  # Should be a list of integers
                 "weights": ["uniform", "distance"],
             }

user_guides/advanced/02_search_space.py

@@ -24,7 +24,7 @@
 knn_module = {
     "module_name": "knn",
     "k": [1, 5, 10, 50],
-    "embedder_name": ["sergeyzh/rubert-tiny-turbo"],
+    "embedder_config": ["sergeyzh/rubert-tiny-turbo"],
 }
 
 # %% [markdown]
@@ -78,7 +78,7 @@
             {
                 "module_name": "retrieval",
                 "k": [10],
-                "embedder_name": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"],
+                "embedder_config": ["avsolatorio/GIST-small-Embedding-v0", "sergeyzh/rubert-tiny-turbo"],
             }
         ],
     },
@@ -90,7 +90,7 @@
             {"module_name": "linear"},
             {
                 "module_name": "dnnc",
-                "cross_encoder_name": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],
+                "embedder_config": ["cross-encoder/ms-marco-MiniLM-L-6-v2"],
                 "k": [1, 3, 5, 10],
             },
         ],