From 5c249cb72729c855ff6d74e5b886b2aea9f3ce23 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 23 May 2025 22:54:46 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/qibotn/backends/quimb.py | 15 +++++++++++---- 1 file changed, 11 insertions(+), 4 deletions(-) diff --git a/src/qibotn/backends/quimb.py b/src/qibotn/backends/quimb.py index 8ce49ce..b3b0b72 100644 --- a/src/qibotn/backends/quimb.py +++ b/src/qibotn/backends/quimb.py @@ -8,6 +8,7 @@ from qibo.result import QuantumState from qibotn.backends.abstract import QibotnBackend from qibotn.result import TensorNetworkResult + class QuimbBackend(QibotnBackend, NumpyBackend): def __init__(self): @@ -108,11 +109,17 @@ class QuimbBackend(QibotnBackend, NumpyBackend): ) frequencies = Counter(circ_quimb.sample(nshots)) if nshots is not None else None - main_frequencies = {state: count for state, count in frequencies.most_common(n=100)} + main_frequencies = { + state: count for state, count in frequencies.most_common(n=100) + } computational_states = [state for state in main_frequencies.keys()] - amplitudes = {state: circ_quimb.amplitude(state) for state in computational_states} - measured_probabilities = {state: abs(amplitude) ** 2 for state, amplitude in amplitudes.items()} - + amplitudes = { + state: circ_quimb.amplitude(state) for state in computational_states + } + measured_probabilities = { + state: abs(amplitude) ** 2 for state, amplitude in amplitudes.items() + } + statevector = circ_quimb.to_dense() if return_array else None return TensorNetworkResult( nqubits=circuit.nqubits,