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) )