'Refactored by Sourcery' (#456)

Co-authored-by: Sourcery AI <>
ElectronicBabylonianLiterature · Sep 12, 2023 · 010b2ac · 010b2ac
1 parent 44afafc
commit 010b2ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/ebl/corpus/infrastructure/corpus_ngram_repository.py b/ebl/corpus/infrastructure/corpus_ngram_repository.py
@@ -56,11 +56,10 @@ def update_ngrams(
         signs_to_exclude: Optional[Sequence[str]] = None,
     ) -> None:
         aggregation = self.aggregate_chapter_ngrams(chapter_id, N, signs_to_exclude)
-        data = next(
+        if data := next(
             self._chapters.aggregate(aggregation, allowDiskUse=True),
             None,
-        )
-        if data:
+        ):
             try:
                 self._ngrams.update_one(
                     {"_id": data["_id"]}, {"$set": {"ngrams": data["ngrams"]}}

diff --git a/ebl/fragmentarium/infrastructure/fragment_ngram_repository.py b/ebl/fragmentarium/infrastructure/fragment_ngram_repository.py
@@ -46,11 +46,10 @@ def update_ngrams(
         signs_to_exclude: Optional[Sequence[str]] = None,
     ) -> None:
         aggregation = self.aggregate_fragment_ngrams(number, N, signs_to_exclude)
-        data = next(
+        if data := next(
             self._fragments.aggregate(aggregation, allowDiskUse=True),
             None,
-        )
-        if data:
+        ):
             try:
                 self._ngrams.update_one(
                     {"_id": data["_id"]}, {"$set": {"ngrams": data["ngrams"]}}