Update utils/block_relation_builder.py

utils/block_relation_builder.py

@@ -1863,8 +1863,10 @@ def classify(line):
 
     # Control Blocks
     # if re.match(r"wait (.+?) seconds", l): return "control_wait", "stack"
-    if re.match(r"wait\s+(\(?\[?.+?\]?\)?)\s*(sec(?:ond)?s?)?$", l): return "control_wait", "stack"
-    if l.startswith("wait until <"): return "control_wait_until", "stack"
+    # if re.match(r"wait\s+(\(?\[?.+?\]?\)?)\s*(sec(?:ond)?s?)?$", l): return "control_wait", "stack"
+    # if l.startswith("wait until <"): return "control_wait_until", "stack"
+    if re.match(r"wait\s+(?!until)(\(?\[?.+?\]?\)?)\s*(sec(?:ond)?s?)?$", l, re.IGNORECASE): return "control_wait", "stack"
+    if re.match(r"wait\s+until\s*<", l, re.IGNORECASE): return "control_wait_until", "stack"
     if l.startswith("repeat ("): return "control_repeat", "c_block"
     if l == "forever": return "control_forever", "c_block"
     if l.startswith("if <") and " then go" in l: return "control_if_else", "c_block"