Update src/qibo/models/circuit.py

Co-authored-by: Renato Mello <[email protected]>
qiboteam · Mar 11, 2024 · 0567eff · 0567eff
1 parent 9d5bf38
commit 0567eff
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/qibo/models/circuit.py b/src/qibo/models/circuit.py
@@ -94,10 +94,11 @@ def moments(self):
         moments = [self.nqubits * [None]]
         moment_index = self.nqubits * [0]
         for gate in self:
-            if not isinstance(gate, gates.CallbackGate):
-                qubits = gate.qubits
-            else:  # special gate acting on all qubits
-                qubits = tuple(range(self.nqubits))
+            qubits = (
+                gate.qubits
+                if not isinstance(gate, gates.CallbackGate)
+                else tuple(range(self.nqubits))  # special gate acting on all qubits
+            )
 
             # calculate moment index for this gate
             idx = max(moment_index[q] for q in qubits)