diff options
Diffstat (limited to 'logical_search.py')
| -rw-r--r-- | logical_search.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/logical_search.py b/logical_search.py index 41ed729..c85b262 100644 --- a/logical_search.py +++ b/logical_search.py @@ -183,7 +183,7 @@ class Corpus(object): def invert_docid_set(self, original: Set[str]) -> Set[str]: """Invert a set of docids.""" - return set([docid for docid in self.documents_by_docid.keys() if docid not in original]) + return {docid for docid in self.documents_by_docid if docid not in original} def get_doc(self, docid: str) -> Optional[Document]: """Given a docid, retrieve the previously added Document.""" |