diff --git a/py_css/models/base.py b/py_css/models/base.py index e04539b..fdc2273 100644 --- a/py_css/models/base.py +++ b/py_css/models/base.py @@ -208,10 +208,6 @@ def gen_context_docs(context: Context) -> Generator[Document, None, None]: for doc in docs: if doc.docno != EMPTY_PLACEHOLDER_DOC.docno: yield doc - i: int = 1 - while True: - yield Document(f"{i}", "") - i += 1 # if in df there are multiple rows that have the same qid and docno, keep the one with the highest score. For the ones removed, add a row each with the EMPTY_PLACEHOLDER_DOC rank_size_per_qid: int = df.groupby("qid").size().max()