Use list comp

microsoft · Aug 29, 2023 · acf7a84 · acf7a84
1 parent 95d6e19
commit acf7a84
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tasks/face_segmentation/snp_test.py b/tasks/face_segmentation/snp_test.py
@@ -51,11 +51,11 @@ def main():
         sys.exit(1)
 
     store = configure_store(aml_config)
-    fully_trained = []
-    for e in store.get_all_status_entities(status='complete'):
-        if metric_key in e:
-            fully_trained += [e]
-
+    fully_trained = [
+        e for e in store.get_all_status_entities(status='complete')
+        if metric_key in e
+    ]
+    
     if len(fully_trained) == 0:
         print(f"No 'complete' models found with required metric '{metric_key}'")
         sys.exit(1)