diff --git a/cirq-core/cirq/transformers/gauge_compiling/idle_moments_gauge.py b/cirq-core/cirq/transformers/gauge_compiling/idle_moments_gauge.py index 8a89e7524ae..f7fea056453 100644 --- a/cirq-core/cirq/transformers/gauge_compiling/idle_moments_gauge.py +++ b/cirq-core/cirq/transformers/gauge_compiling/idle_moments_gauge.py @@ -187,7 +187,9 @@ def __call__( for q in op.qubits: active_moments[q].append((m_id, is_mergable)) - single_qubit_moments = [{q: op for op in m if len(op.qubits) == 1} for m in circuit] + single_qubit_moments = [ + {op.qubits[0]: op for op in m if len(op.qubits) == 1} for m in circuit + ] non_single_qubit_moments = [[op for op in m if len(op.qubits) != 1] for m in circuit] for q, active in active_moments.items():