From 585e758c4ef1f9f39fa6c68eed36611dd1256e3b Mon Sep 17 00:00:00 2001 From: Andrea Pasquale Date: Mon, 23 Sep 2024 19:33:30 +0400 Subject: [PATCH] refactor: improve readability Co-authored-by: Alessandro Candido --- src/qibocal/protocols/state_tomography.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/qibocal/protocols/state_tomography.py b/src/qibocal/protocols/state_tomography.py index 61dcde961..48b13cf1a 100644 --- a/src/qibocal/protocols/state_tomography.py +++ b/src/qibocal/protocols/state_tomography.py @@ -147,8 +147,7 @@ def _fit(data: StateTomographyData) -> StateTomographyResults: circuit.density_matrix = True total_density_matrix = NumpyBackend().execute_circuit(circuit=circuit).state() for i, qubit in enumerate(data.targets): - traced_qubits = list(range(len(data.qubits))) - traced_qubits.remove(i) + traced_qubits = [q for q in range(len(data.qubits)) if q != i] target_density_matrix = NumpyBackend().partial_trace_density_matrix( total_density_matrix, traced_qubits, len(data.qubits) )