fix: Improve imports for new rankers (#5696)

* Proper imports for new rankers * Small fix
deepset-ai · Aug 31, 2023 · 11afa0a · 11afa0a
1 parent 7d39184
commit 11afa0a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 5 deletions.
diff --git a/examples/talk_to_website.py b/examples/talk_to_website.py
@@ -5,7 +5,7 @@
 from haystack import Pipeline
 from haystack.document_stores import InMemoryDocumentStore
 from haystack.nodes import PromptNode, PromptTemplate, TopPSampler
-from haystack.nodes.ranker.lost_in_the_middle import LostInTheMiddleRanker
+from haystack.nodes.ranker import LostInTheMiddleRanker
 from haystack.nodes.retriever.web import WebRetriever
 
 search_key = os.environ.get("SERPERDEV_API_KEY")

diff --git a/examples/web_lfqa_with_rankers.py b/examples/web_lfqa_with_rankers.py
@@ -4,9 +4,8 @@
 
 from haystack import Pipeline
 from haystack.nodes import PromptNode, PromptTemplate, TopPSampler
-from haystack.nodes.ranker.diversity import DiversityRanker
-from haystack.nodes.ranker.lost_in_the_middle import LostInTheMiddleRanker
-from haystack.nodes.retriever.web import WebRetriever
+from haystack.nodes.ranker import DiversityRanker, LostInTheMiddleRanker
+from haystack.nodes.retriever import WebRetriever
 
 search_key = os.environ.get("SERPERDEV_API_KEY")
 if not search_key:

diff --git a/haystack/nodes/ranker/__init__.py b/haystack/nodes/ranker/__init__.py
@@ -1,3 +1,6 @@
 from haystack.nodes.ranker.base import BaseRanker
 from haystack.nodes.ranker.sentence_transformers import SentenceTransformersRanker
 from haystack.nodes.ranker.cohere import CohereRanker
+from haystack.nodes.ranker.lost_in_the_middle import LostInTheMiddleRanker
+from haystack.nodes.ranker.diversity import DiversityRanker
+from haystack.nodes.ranker.recentness_ranker import RecentnessRanker
diff --git a/haystack/nodes/ranker/diversity.py b/haystack/nodes/ranker/diversity.py
@@ -2,7 +2,7 @@
 from pathlib import Path
 from typing import List, Literal, Optional, Union
 
-from haystack.nodes import BaseRanker
+from haystack.nodes.ranker.base import BaseRanker
 from haystack.schema import Document
 from haystack.lazy_imports import LazyImport