optimized the for loop for extensions and targets usage

riscv · Oct 27, 2024 · 837fbba · 837fbba
1 parent 1166a7d
commit 837fbba
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/parse.py b/parse.py
@@ -25,7 +25,8 @@
     print(f"Running with args : {sys.argv}")
 
     extensions = sys.argv[1:]
-    for i in [
+
+    targets = {
         "-c",
         "-chisel",
         "-go",
@@ -34,9 +35,9 @@
         "-rust",
         "-spinalhdl",
         "-sverilog",
-    ]:
-        if i in extensions:
-            extensions.remove(i)
+    }
+
+    extensions = [ext for ext in extensions if ext not in targets]
     print(f"Extensions selected : {extensions}")
 
     include_pseudo = False