Utilize list comprehension

KIT-TVA · Aug 28, 2024 · 6bce804 · 6bce804
1 parent 7f2fa57
commit 6bce804
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cfmtoolbox/plugins/random_sampling.py b/cfmtoolbox/plugins/random_sampling.py
@@ -13,9 +13,9 @@ def random_sampling(model: CFM | None, amount: int = 1) -> CFM | None:
         print("No model loaded.")
         return None
 
-    all_samples = []
-    for _ in range(amount):
-        all_samples.append(asdict(RandomSampler(model).random_sampling()))
+    all_samples = [
+        asdict(RandomSampler(model).random_sampling()) for _ in range(amount)
+    ]
 
     print(json.dumps(all_samples, indent=2))