Merge branch 'dev' into feat/async-augmentations

Author: Darinochka

.github/workflows/generate-schema.yaml

@@ -3,7 +3,7 @@ name: Generate JSON Schema
 on:
   push:
     branches:
-      - main
+      - dev
 
 permissions:
   contents: write

autointent/modules/embedding/_logreg.py

@@ -77,8 +77,8 @@ def __init__(
     def from_context(
         cls,
         context: Context,
-        cv: int,
         embedder_name: str,
+        cv: int = 3,
     ) -> "LogregAimedEmbedding":
         """
         Create a LogregAimedEmbedding instance using a Context object.
@@ -89,8 +89,8 @@ def from_context(
         :return: Initialized LogregAimedEmbedding instance.
         """
         return cls(
-            cv=cv,
             embedder_name=embedder_name,
+            cv=cv,
             embedder_device=context.get_device(),
             embedder_batch_size=context.get_batch_size(),
             embedder_max_length=context.get_max_length(),

tests/generation/utterances/test_basic_synthesizer.py

@@ -51,6 +51,7 @@ def test_on_dataset(dataset):
     assert n_before + len(new_samples) == n_after
     assert len(new_samples) == len(dataset.intents)
 
+
 def test_on_dataset_async(dataset):
     mock_llm = AsyncMock()
     mock_llm.get_chat_completion_async.return_value = "1. LLM answer"

tests/generation/utterances/test_evolver.py

@@ -33,6 +33,7 @@ def test_on_dataset(dataset):
     assert n_before + len(new_samples) == n_after
     assert len(new_samples) == n_before
 
+
 def test_on_dataset_evolver_async(dataset):
     mock_llm = AsyncMock()
     mock_llm.get_chat_completion_async.return_value = "LLM answer"
@@ -88,4 +89,4 @@ def test_on_dataset_evolver_async_with_batch_size(dataset):
     )
 
     assert len(new_samples) == len(dataset[split_name])
-    assert all(sample.utterance == "LLM answer" for sample in new_samples)
+    assert all(sample.utterance == "LLM answer" for sample in new_samples)