X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=logical_search.py;h=ef55a2bbdcbb5baafa5c29a53fc404fcfb4701bb;hb=fdfbe2d80c21f36083065899c37f777e6cf2a138;hp=41ed729fc15df80a4477d3094e4dc980869741aa;hpb=244e8476c95d14a480be7160042b2b27b693ca63;p=python_utils.git

diff --git a/logical_search.py b/logical_search.py
index 41ed729..ef55a2b 100644
--- a/logical_search.py
+++ b/logical_search.py
@@ -5,7 +5,6 @@ corpus of documents.  The corpus is held in memory for fast
 searching."""
 
 from __future__ import annotations
-
 import enum
 import sys
 from collections import defaultdict
@@ -183,7 +182,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."""