Merge pull request #245 from tosemml/patch-1

Refactoring
microsoft · Sep 15, 2023 · 0f55ccb · 0f55ccb
2 parents 3503870 + acf7a84
commit 0f55ccb
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)