Fix py3.9

QuTech-Delft · Sep 30, 2024 · dd38505 · dd38505
1 parent 2069dec
commit dd38505
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/opensquirrel/writer/writer.py b/opensquirrel/writer/writer.py
@@ -59,11 +59,11 @@ def visit_gate(self, gate: Gate) -> None:
                 gate_name = gate_name.replace("\n", "")
             self.output += f"{gate_name}\n"
             return
-        if any(not isinstance(arg, QubitLike) for arg in gate.arguments):  # type: ignore
-            params = [arg.accept(self) for arg in gate.arguments if not isinstance(arg, QubitLike)]  # type: ignore
+        if any(not isinstance(arg, QubitLike.__args__) for arg in gate.arguments):  # type: ignore
+            params = [arg.accept(self) for arg in gate.arguments if not isinstance(arg, QubitLike.__args__)]  # type: ignore
             gate_name += f"({', '.join(params)})"
 
-        qubit_args = (Qubit(arg).accept(self) for arg in gate.arguments if isinstance(arg, QubitLike))  # type: ignore
+        qubit_args = (Qubit(arg).accept(self) for arg in gate.arguments if isinstance(arg, QubitLike.__args__))  # type: ignore
 
         self.output += f"{gate_name} {', '.join(qubit_args)}\n"