fix for non-groups

tum-ei-eda · Jun 27, 2022 · cf20fc0 · cf20fc0 · PhilippvK · Jun 27, 2022
1 parent a995f01
commit cf20fc0
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mlonmcu/models/lookup.py b/mlonmcu/models/lookup.py
@@ -320,8 +320,10 @@ def apply_modelgroups(models, context=None):
                 )
             groups[group.name] = group.models
 
-    models = [m for model in models for m in groups[model]]
-    return list(dict.fromkeys(models))  # Drop duplicates
+    out_models = []
+    for model in models:
+        out_models.extend(groups.get(model, [model]))
+    return list(dict.fromkeys(out_models))  # Drop duplicates
 
 
 def map_frontend_to_model(model, frontends, backend=None):