shorten code

RagnarGrootKoerkamp · Nov 17, 2024 · 0dbd193 · 0dbd193
1 parent 2c5aa25
commit 0dbd193
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/bin/generate.py b/bin/generate.py
@@ -1517,16 +1517,13 @@ def parse_count(yaml, warn_for=None):
                         color_type=MessageType.ERROR,
                     )
                 return 1000
-            if count > 100:
-                if warn_for is not None:
-                    message(
-                        f'Found large count: {count}.',
-                        'generators.yaml',
-                        warn_for,
-                        color_type=MessageType.LOG,
-                    )
-                return count
-
+            if count > 100 and warn_for is not None:
+                message(
+                    f'Found large count: {count}.',
+                    'generators.yaml',
+                    warn_for,
+                    color_type=MessageType.LOG,
+                )
             return count
 
         # Count the number of testcases in the given directory yaml.