fix(tests): flair inference tests

tests/test_flair_inference.py

@@ -12,11 +12,8 @@
 
 from embeddings.data.data_loader import HuggingFaceDataLoader
 from embeddings.data.dataset import Dataset
-from embeddings.embedding.flair_embedding import FlairDocumentPoolEmbedding
-from embeddings.embedding.flair_loader import (
-    FlairDocumentPoolEmbeddingLoader,
-    FlairWordEmbeddingLoader,
-)
+from embeddings.embedding.auto_flair import AutoFlairDocumentEmbedding
+from embeddings.embedding.flair_loader import FlairWordEmbeddingLoader
 from embeddings.evaluator.sequence_labeling_evaluator import SequenceLabelingEvaluator
 from embeddings.evaluator.text_classification_evaluator import TextClassificationEvaluator
 from embeddings.model.flair_model import FlairModel
@@ -59,10 +56,9 @@ def text_classification_pipeline(
     )
     data_loader = HuggingFaceDataLoader()
     transformation = ClassificationCorpusTransformation("text", "target").then(
-        DownsampleFlairCorpusTransformation(*(0.005, 0.01, 0.01), stratify=False)
+        DownsampleFlairCorpusTransformation(*(0.01, 0.01, 0.01), stratify=False)
     )
-    embedding_loader = FlairDocumentPoolEmbeddingLoader("clarin-pl/word2vec-kgr10", "")
-    embedding = embedding_loader.get_embedding(FlairDocumentPoolEmbedding)
+    embedding = AutoFlairDocumentEmbedding.from_hub("allegro/herbert-base-cased")
     task = TextClassification(output_path.name, task_train_kwargs={"max_epochs": 1})
     model = FlairModel(embedding, task)
     evaluator = TextClassificationEvaluator()