simplify ntags expression

Co-authored-by: Andreas Klöckner <[email protected]>
inducer · Jun 12, 2024 · 3b50d35 · 3b50d35
1 parent 4d96e18
commit 3b50d35
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pytato/array.py b/pytato/array.py
@@ -465,9 +465,9 @@ def _is_eq_valid(self) -> bool:
         def __attrs_post_init__(self) -> None:
             if _ENABLE_TRACEBACK_TAG:
                 from pytato.tags import CreatedAt
-                ntags = sum(
-                    1 if isinstance(tag, CreatedAt) else 0
-                    for tag in self.non_equality_tags)
+                ntags = len(
+                    tag for tag in self.non_equality_tags
+                    if isinstance(tag, CreatedAt))
                 if ntags < 1:
                     raise AssertionError("Array has no CreatedAt tag.")
                 elif ntags > 1: