Treat COMPARE_OP_*_JUMP as superinstructions

brandtbucher · Jun 17, 2022 · 67c8382 · 67c8382
1 parent 8e3071b
commit 67c8382
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/specialist.py b/specialist.py
@@ -99,7 +99,10 @@ def stderr(*args: object) -> None:
 
 def is_superinstruction(instruction: dis.Instruction) -> bool:
     """Check if an instruction is a superinstruction."""
-    return "__" in instruction.opname
+    opname = instruction.opname
+    return "__" in opname or (
+        opname.startswith("COMPARE_OP_") and opname.endswith("_JUMP")
+    )
 
 
 def score_instruction(