fix import issues

deeppavlov · Jan 27, 2025 · 2cefd7f · 2cefd7f
1 parent 9200e05
commit 2cefd7f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/autointent/__init__.py b/autointent/__init__.py
@@ -1,13 +1,14 @@
 """This is AutoIntent API reference."""
 
-from ._dataset import Dataset
-from ._embedder import Embedder
-from ._hash import Hasher
 from ._logging import setup_logging
-from ._pipeline import Pipeline
 from ._ranker import Ranker
+from ._embedder import Embedder
 from ._vector_index import VectorIndex
+from ._dataset import Dataset
+from ._hash import Hasher
 from .context import Context, load_dataset
+from ._pipeline import Pipeline
+
 
 __all__ = [
     "Context",

diff --git a/tests/generation/test_description_generation.py b/tests/generation/test_description_generation.py
@@ -4,13 +4,13 @@
 import pytest
 
 from autointent import Dataset
-from autointent.generation.description_generation import (
+from autointent.generation.intents.description_generation import (
     create_intent_description,
     enhance_dataset_with_descriptions,
     generate_intent_descriptions,
     group_utterances_by_label,
 )
-from autointent.generation.prompt_scheme import PromptDescription
+from autointent.generation.intents.prompt_scheme import PromptDescription
 from autointent.schemas import Intent, Sample