From 6ab8edf3ab4b2d941c42156bc966b30f32cf6a2e Mon Sep 17 00:00:00 2001 From: mert Date: Wed, 14 Dec 2022 11:20:02 +0300 Subject: [PATCH] fix num_sentences not used on __call__ --- summarizer/cluster_features.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/summarizer/cluster_features.py b/summarizer/cluster_features.py index 3759458..a7cf3e8 100644 --- a/summarizer/cluster_features.py +++ b/summarizer/cluster_features.py @@ -162,4 +162,4 @@ def __call__(self, ratio: float = 0.1, num_sentences: int = None) -> List[int]: :param num_sentences: Number of sentences. Overrides ratio. :return: Sentences index that qualify for summary. """ - return self.cluster(ratio) + return self.cluster(ratio, num_sentences)