Skip to content
Snippets Groups Projects
Commit 525c24ba authored by vvye's avatar vvye
Browse files

Bring rounding back

parent c1e8ec8e
No related branches found
No related tags found
No related merge requests found
......@@ -17,9 +17,9 @@ def summarize(sentences, vectorizer, keywords, by_tokens, num_sentences, num_tok
# (either checking for number of sentences or number of tokens)
def constraint():
if by_tokens:
return sum([len(sentences[i]['text'].split()) for i in selected_indices]) < num_tokens
return sum([len(sentences[i]['text'].split()) for i in selected_indices]) < round(num_tokens)
else:
return len(selected_indices) < num_sentences
return len(selected_indices) < round(num_sentences)
while remaining_indices and constraint():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment