Fixing various typing issues

zapata-engineering · Jan 20, 2024 · abefda2 · abefda2
1 parent 4af3cdc
commit abefda2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/orquestra/quantum/circuits/_gates.py b/src/orquestra/quantum/circuits/_gates.py
@@ -218,8 +218,8 @@ def bind(self, symbols_map) -> "MatrixFactoryGate":
     def replace_params(self, new_params: Tuple[Parameter, ...]) -> "MatrixFactoryGate":
         return replace(self, params=new_params)
 
-    def controlled(self, num_controlled_qubits: int) -> Gate:
-        return ControlledGate(self, num_controlled_qubits)
+    def controlled(self, num_control_qubits: int) -> Gate:
+        return ControlledGate(self, num_control_qubits)
 
     @property
     def dagger(self) -> Union["MatrixFactoryGate", Gate]:

diff --git a/src/orquestra/quantum/decompositions/_orquestra_decompositions.py b/src/orquestra/quantum/decompositions/_orquestra_decompositions.py
@@ -31,7 +31,7 @@ def production(self, operation: GateOperation) -> Iterable[GateOperation]:
         def preprocess_gate(gate):
             return (
                 gate.controlled(operation.gate.num_control_qubits)
-                if operation.gate.name == "Control"
+                if isinstance(operation.gate, ControlledGate)
                 else gate
             )