core/database/utils/count_documents.py

@@ -6,16 +6,16 @@ from pymongo.collection import Collection
 
 def count_documents(
     collection: Collection,
-    has_annotation: bool = False,
+    only_with_annotation: bool = False,
 ) -> int:
     """
     Get the total number of documents
     in database that matches the filters.
     """
     assert isinstance(collection, Collection)
-    assert isinstance(has_annotation, bool)
+    assert isinstance(only_with_annotation, bool)
     # build query
     query = {}
-    if has_annotation:
+    if only_with_annotation:
         query['annotation'] = {'$ne': None}
     return collection.count_documents(filter=query)