Allow lists of wire

qermit/taskgraph/mittask.py

@@ -57,6 +57,23 @@ class IOTask(Enum):
     Dict,
 ]
 
+WireList = Union[
+    List[PostselectMgr],
+    List[CircuitShots],
+    List[Circuit],
+    List[BackendResult],
+    List[ResultHandle],
+    List[AnsatzCircuit],
+    List[ObservableExperiment],
+    List[int],
+    List[float],
+    List[bool],
+    List[str],
+    List[QubitPauliOperator],
+    List[Dict[Qubit, Bit]],
+    List[Dict],
+]
+
 
 class MitTask:
     """
@@ -100,7 +117,7 @@ def n_in_wires(self):
     def n_out_wires(self):
         return self._n_out_wires
 
-    def __call__(self, input_wires: Tuple[List[Wire], ...]) -> Tuple[List[Wire], ...]:
+    def __call__(self, input_wires: Tuple[WireList, ...]) -> Tuple[WireList, ...]:
         return self.run(*input_wires)
 
     def __str__(self):