Typing improvements.

CQCL · Sep 11, 2024 · 7ec42be · 7ec42be
1 parent 35f7f6f
commit 7ec42be
Showing 1 changed file with 11 additions and 11 deletions.
diff --git a/pytket/pytket/qasm/qasm.py b/pytket/pytket/qasm/qasm.py
@@ -1290,7 +1290,7 @@ def make_params_str(params: Optional[List[Union[float, Expr]]]) -> str:
     return s
 
 
-def make_args_str(args: List[UnitID]) -> str:
+def make_args_str(args: Sequence[UnitID]) -> str:
     s = ""
     for i in range(len(args)):
         s += f"{args[i]}"
@@ -1425,7 +1425,7 @@ def write_params(self, params: Optional[List[Union[float, Expr]]]) -> None:
         params_str = make_params_str(params)
         self.strings.add_string(params_str)
 
-    def write_args(self, args: List[UnitID]) -> None:
+    def write_args(self, args: Sequence[UnitID]) -> None:
         args_str = make_args_str(args)
         self.strings.add_string(args_str)
 
@@ -1579,7 +1579,7 @@ def remove_unused_predicate(self, pred_label: int) -> bool:
         self.strings.del_string(pred_label)
         return True
 
-    def add_conditional(self, op: Conditional, args: List[UnitID]) -> None:
+    def add_conditional(self, op: Conditional, args: Sequence[UnitID]) -> None:
         control_bits = args[: op.width]
         if op.width == 1 and hqs_header(self.header):
             variable = str(control_bits[0])
@@ -1735,11 +1735,11 @@ def add_wasm(self, op: WASMOp, args: List[Bit]) -> None:
         for variable in outputs:
             self.mark_as_written(label, variable)
 
-    def add_measure(self, args: List[UnitID]) -> None:
+    def add_measure(self, args: Sequence[UnitID]) -> None:
         label = self.strings.add_string(f"measure {args[0]} -> {args[1]};\n")
         self.mark_as_written(label, f"{args[1]}")
 
-    def add_zzphase(self, param: Union[float, Expr], args: List[UnitID]) -> None:
+    def add_zzphase(self, param: Union[float, Expr], args: Sequence[UnitID]) -> None:
         # as op.params returns reduced parameters, we can assume
         # that 0 <= param < 4
         if param > 1:
@@ -1753,7 +1753,7 @@ def add_zzphase(self, param: Union[float, Expr], args: List[UnitID]) -> None:
         self.write_params([param])
         self.write_args(args)
 
-    def add_data(self, op: BarrierOp, args: List[UnitID]) -> None:
+    def add_data(self, op: BarrierOp, args: Sequence[UnitID]) -> None:
         if op.data == "":
             opstr = _tk_to_qasm_noparams[OpType.Barrier]
         else:
@@ -1762,18 +1762,18 @@ def add_data(self, op: BarrierOp, args: List[UnitID]) -> None:
         self.strings.add_string(" ")
         self.write_args(args)
 
-    def add_gate_noparams(self, op: Op, args: List[UnitID]) -> None:
+    def add_gate_noparams(self, op: Op, args: Sequence[UnitID]) -> None:
         self.strings.add_string(_tk_to_qasm_noparams[op.type])
         self.strings.add_string(" ")
         self.write_args(args)
 
-    def add_gate_params(self, op: Op, args: List[UnitID]) -> None:
+    def add_gate_params(self, op: Op, args: Sequence[UnitID]) -> None:
         optype, params = _get_optype_and_params(op)
         self.strings.add_string(_tk_to_qasm_params[optype])
         self.write_params(params)
         self.write_args(args)
 
-    def add_extra_noparams(self, op: Op, args: List[UnitID]) -> Tuple[str, str]:
+    def add_extra_noparams(self, op: Op, args: Sequence[UnitID]) -> Tuple[str, str]:
         optype = op.type
         opstr = _tk_to_qasm_extra_noparams[optype]
         gatedefstr = ""
@@ -1783,7 +1783,7 @@ def add_extra_noparams(self, op: Op, args: List[UnitID]) -> Tuple[str, str]:
         mainstr = opstr + " " + make_args_str(args)
         return gatedefstr, mainstr
 
-    def add_extra_params(self, op: Op, args: List[UnitID]) -> Tuple[str, str]:
+    def add_extra_params(self, op: Op, args: Sequence[UnitID]) -> Tuple[str, str]:
         optype, params = _get_optype_and_params(op)
         assert params is not None
         opstr = _tk_to_qasm_extra_params[optype]
@@ -1796,7 +1796,7 @@ def add_extra_params(self, op: Op, args: List[UnitID]) -> Tuple[str, str]:
         mainstr = opstr + make_params_str(params) + make_args_str(args)
         return gatedefstr, mainstr
 
-    def add_op(self, op: Op, args: List[UnitID]) -> None:
+    def add_op(self, op: Op, args: Sequence[UnitID]) -> None:
         optype, _params = _get_optype_and_params(op)
         if optype == OpType.RangePredicate:
             assert isinstance(op, RangePredicateOp)