From bb95e78132fa9b9cfe0fb76496bf6844ccac4315 Mon Sep 17 00:00:00 2001 From: Brian Bjarke Jensen Date: Sun, 31 Mar 2024 12:20:03 +0200 Subject: [PATCH] renamed argument for clarity --- core/database/utils/count_documents.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/core/database/utils/count_documents.py b/core/database/utils/count_documents.py index b254aef..53a4a1c 100755 --- a/core/database/utils/count_documents.py +++ b/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)