diff --git a/haystack/nodes/__init__.py b/haystack/nodes/__init__.py index 062acfb26..a969e245b 100644 --- a/haystack/nodes/__init__.py +++ b/haystack/nodes/__init__.py @@ -26,7 +26,14 @@ from haystack.nodes.prompt import PromptNode, PromptTemplate, PromptModel, BaseO from haystack.nodes.prompt.invocation_layer import PromptModelInvocationLayer from haystack.nodes.query_classifier import TransformersQueryClassifier from haystack.nodes.question_generator import QuestionGenerator -from haystack.nodes.ranker import BaseRanker, SentenceTransformersRanker, CohereRanker +from haystack.nodes.ranker import ( + BaseRanker, + SentenceTransformersRanker, + CohereRanker, + LostInTheMiddleRanker, + DiversityRanker, + RecentnessRanker, +) from haystack.nodes.reader import BaseReader, FARMReader, TransformersReader, TableReader, RCIReader from haystack.nodes.retriever import ( BaseRetriever, diff --git a/releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml b/releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml new file mode 100644 index 000000000..b77bdf0cc --- /dev/null +++ b/releasenotes/notes/add-new-rankers-schema-generation-7ad92fd5c5de8937.yaml @@ -0,0 +1,5 @@ +--- +fixes: + - | + Adds LostInTheMiddleRanker, DiversityRanker, and RecentnessRanker to `haystack/nodes/__init__.py` and thus + ensures that they are included in JSON schema generation.