LAVA Helper: Fix logic to filter device types

Use list comprehension to filter device types in LAVA Helper.
Directly removing items during iteration will lead to some items not
iterated as expected because of the dynamic index.

Signed-off-by: Xinyu Zhang <xinyu.zhang@arm.com>
Change-Id: Id1b557f57567eca1446ffb39563debeae787cb45
diff --git a/lava_helper/lava_create_jobs.py b/lava_helper/lava_create_jobs.py
index 6c29f4d..71c6e53 100755
--- a/lava_helper/lava_create_jobs.py
+++ b/lava_helper/lava_create_jobs.py
@@ -191,13 +191,10 @@
     user_args.template_dir = "jinja2_templates"
     config_keys = list(lava_gen_config_map.keys())
     if user_args.fvp_only:
-        for key in config_keys:
-            if "fvp" not in key:
-                config_keys.remove(key)
+        config_keys = [key for key in config_keys if "fvp" in key]
     if user_args.physical_board_only:
-        for key in config_keys:
-            if "fvp" in key or "qemu" in key:
-                config_keys.remove(key)
+        config_keys = [key for key in config_keys
+                       if "fvp" not in key and "qemu" not in key]
     if user_args.config_key:
         config_keys = [user_args.config_key]
     for config_key in config_keys: