From 1c043d95f5c1ced043bf7fc02e9fc4167e608d5d Mon Sep 17 00:00:00 2001 From: Liwei Yang Date: Thu, 2 Nov 2023 17:17:56 +0800 Subject: [PATCH] Minor formatting update --- src/qibotn/quimb.py | 4 ++-- tests/test_qasm_quimb_backend.py | 6 ++++-- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/qibotn/quimb.py b/src/qibotn/quimb.py index 5f5721b..50c25ce 100644 --- a/src/qibotn/quimb.py +++ b/src/qibotn/quimb.py @@ -3,7 +3,7 @@ import quimb.tensor as qtn from qibo.models import Circuit as QiboCircuit -def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd', +def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd', cutoff=1e-6, cutoff_mode='abs'): nqubits = circuit.nqubits gate_opt = {} @@ -12,7 +12,7 @@ def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd', gate_opt["method"] = method gate_opt["cutoff"] = cutoff gate_opt["cutoff_mode"] = cutoff_mode - else: + else: tncirc = qtn.Circuit(nqubits, psi0=psi0) for gate in circuit.queue: diff --git a/tests/test_qasm_quimb_backend.py b/tests/test_qasm_quimb_backend.py index 3fd756e..f46fbc6 100644 --- a/tests/test_qasm_quimb_backend.py +++ b/tests/test_qasm_quimb_backend.py @@ -10,7 +10,8 @@ from qibo.models import QFT def create_init_state(nqubits): - init_state = np.random.random(2**nqubits) + 1j * np.random.random(2**nqubits) + init_state = np.random.random(2**nqubits) + \ + 1j * np.random.random(2**nqubits) init_state = init_state / np.sqrt((np.abs(init_state) ** 2).sum()) return init_state @@ -42,7 +43,8 @@ def test_eval(nqubits: int, tolerance: float): init_state_tn = copy.deepcopy(init_state) # Test qibo - qibo.set_backend(backend=config.qibo.backend, platform=config.qibo.platform) + qibo.set_backend(backend=config.qibo.backend, + platform=config.qibo.platform) qibo_time, (qibo_circ, result_sv) = time( lambda: qibo_qft(nqubits, init_state, swaps=True) )