Skip to content

Commit 6787ad2

Browse files
authored
fix: Improve imports for new rankers (#5696)
* Proper imports for new rankers * Small fix
1 parent 55b10a3 commit 6787ad2

File tree

4 files changed

+7
-5
lines changed

4 files changed

+7
-5
lines changed

examples/talk_to_website.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
from haystack import Pipeline
66
from haystack.document_stores import InMemoryDocumentStore
77
from haystack.nodes import PromptNode, PromptTemplate, TopPSampler
8-
from haystack.nodes.ranker.lost_in_the_middle import LostInTheMiddleRanker
8+
from haystack.nodes.ranker import LostInTheMiddleRanker
99
from haystack.nodes.retriever.web import WebRetriever
1010

1111
search_key = os.environ.get("SERPERDEV_API_KEY")

examples/web_lfqa_with_rankers.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,9 +4,8 @@
44

55
from haystack import Pipeline
66
from haystack.nodes import PromptNode, PromptTemplate, TopPSampler
7-
from haystack.nodes.ranker.diversity import DiversityRanker
8-
from haystack.nodes.ranker.lost_in_the_middle import LostInTheMiddleRanker
9-
from haystack.nodes.retriever.web import WebRetriever
7+
from haystack.nodes.ranker import DiversityRanker, LostInTheMiddleRanker
8+
from haystack.nodes.retriever import WebRetriever
109

1110
search_key = os.environ.get("SERPERDEV_API_KEY")
1211
if not search_key:

haystack/nodes/ranker/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
11
from haystack.nodes.ranker.base import BaseRanker
22
from haystack.nodes.ranker.sentence_transformers import SentenceTransformersRanker
33
from haystack.nodes.ranker.cohere import CohereRanker
4+
from haystack.nodes.ranker.lost_in_the_middle import LostInTheMiddleRanker
5+
from haystack.nodes.ranker.diversity import DiversityRanker
6+
from haystack.nodes.ranker.recentness_ranker import RecentnessRanker

haystack/nodes/ranker/diversity.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
from pathlib import Path
33
from typing import List, Literal, Optional, Union
44

5-
from haystack.nodes import BaseRanker
5+
from haystack.nodes.ranker.base import BaseRanker
66
from haystack.schema import Document
77
from haystack.lazy_imports import LazyImport
88

0 commit comments

Comments
 (0)