diff --git a/qsimcirq/qsim_circuit.py b/qsimcirq/qsim_circuit.py index a5b4ae277..ac14af36f 100644 --- a/qsimcirq/qsim_circuit.py +++ b/qsimcirq/qsim_circuit.py @@ -306,9 +306,7 @@ def add_op_to_circuit( time, qsim_qubits, qsim_gate._diag_angles_radians, circuit ) elif gate_kind == qsim.kMatrixGate: - m = [ - val for i in list(cirq.unitary(qsim_gate).flat) for val in [i.real, i.imag] - ] + m = cirq.unitary(qsim_gate).ravel().view(np.float64).tolist() if isinstance(circuit, qsim.Circuit): qsim.add_matrix_gate(time, qsim_qubits, m, circuit) else: