fix after pull dev

Author: SeBorgey

autointent/generation/utterances/balancer.py

@@ -7,7 +7,7 @@
 
 from autointent import Dataset
 from autointent.custom_types import Split
-from autointent.generation.utterances.basic.chat_template import BaseSynthesizer
+from autointent.generation.utterances.basic.chat_templates._base import BaseChatTemplate
 from autointent.generation.utterances.basic.utterance_generator import UtteranceGenerator
 from autointent.generation.utterances.generator import Generator
 
@@ -20,7 +20,7 @@ class DatasetBalancer:
     def __init__(
         self,
         generator: Generator,
-        prompt_maker: BaseSynthesizer,
+        prompt_maker: BaseChatTemplate,
         async_mode: bool = False,
         max_samples_per_class: int | None = None,
     ) -> None:

tests/generation/utterances/test_balancer.py

@@ -8,7 +8,7 @@
 from autointent import Dataset
 from autointent.custom_types import Split
 from autointent.generation.utterances import DatasetBalancer, Generator
-from autointent.generation.utterances.basic.chat_template import SynthesizerChatTemplate
+from autointent.generation.utterances.basic.chat_templates._synthesizer_en import EnglishSynthesizerTemplate
 
 logger = logging.getLogger(__name__)
 
@@ -76,7 +76,7 @@ def test_real_balancer():
         ],
     }
     dataset = Dataset.from_dict(test_data)
-    template = SynthesizerChatTemplate(dataset, split="train")
+    template = EnglishSynthesizerTemplate(dataset, split="train")
     generator = Generator()
     evolutions = template
     balancer = DatasetBalancer(generator=generator, prompt_maker=evolutions, max_samples_per_class=3, async_mode=False)