From 53382b8184761409e81d6be4fbd17b3be3931ace Mon Sep 17 00:00:00 2001 From: Liwei Yang Date: Wed, 15 Feb 2023 17:28:06 +0800 Subject: [PATCH] Use autopep8 for PEP-8 formatting --- src/qibotn/qasm_quimb.py | 22 ++++++++++++++-------- tests/test_qasm_quimb_backend.py | 18 ++++++++++-------- 2 files changed, 24 insertions(+), 16 deletions(-) diff --git a/src/qibotn/qasm_quimb.py b/src/qibotn/qasm_quimb.py index a7fadc8..a552ad0 100644 --- a/src/qibotn/qasm_quimb.py +++ b/src/qibotn/qasm_quimb.py @@ -26,7 +26,8 @@ def get_gate_params(operation): qbit_no.insert(0, "T") elif "cu1" in operation: lambda_ = float( - ".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) + ".".join(re.findall( + r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ) qbit_no = re.findall(r"\d+", operation.split(" ")[1]) qbit_no = [int(x) for x in qbit_no] @@ -72,32 +73,37 @@ def get_gate_params(operation): qbit_no.insert(0, "CCZ") elif " rx " in operation: theta = float( - ".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) + ".".join(re.findall( + r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ) qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no[0:0] = ["RX", theta] elif "^ry " in operation: theta = float( - ".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) + ".".join(re.findall( + r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ) qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no[0:0] = ["RY", theta] elif "^rz " in operation: theta = float( - ".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) + ".".join(re.findall( + r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ) qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no[0:0] = ["RZ", theta] elif "^rzz " in operation: theta = float( - ".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) + ".".join(re.findall( + r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ) qbit_no = re.findall(r"\d+", operation.split(" ")[1]) qbit_no = [int(x) for x in qbit_no] qbit_no[0:0] = ["RZZ", theta] elif "^u1 " in operation: lambda_ = float( - ".".join(re.findall(r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) + ".".join(re.findall( + r"\b\d+(?:[Ee][+-]?\d+)?", operation.split(" ")[0])) ) qbit_no = [int(re.findall(r"\d+", operation)[0])] qbit_no[0:0] = ["U1", lambda_] @@ -182,7 +188,7 @@ def init_state_tn(nqubits, init_state_sv, tn_lib="quimb"): def tn_circ_eval(nqubits, qasm_circ, init_state, swaps=True, tn_lib="quimb", - backend='numpy'): + backend='numpy'): if tn_lib == "quimb": circ_quimb = qasm_QFT(nqubits, qasm_circ, swaps, psi0=init_state) @@ -202,4 +208,4 @@ def eval_QI_qft(nqubits, qasm_circ, init_state, backend="numpy", swaps=True): amplitudes = tn_circ_eval(nqubits=nqubits, qasm_circ=qasm_circ, init_state=init_state_mps, swaps=swaps, tn_lib="quimb") - return amplitudes \ No newline at end of file + return amplitudes diff --git a/tests/test_qasm_quimb_backend.py b/tests/test_qasm_quimb_backend.py index dfe83c9..450104b 100644 --- a/tests/test_qasm_quimb_backend.py +++ b/tests/test_qasm_quimb_backend.py @@ -11,7 +11,8 @@ import config def init_state_sv(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()) # An unmodified init_state has to be converted to tn format init_state_for_tn = copy.deepcopy(init_state) @@ -32,11 +33,11 @@ def test_eval(nqubits: int): init_state_qibo, init_state_for_tn = init_state_sv(nqubits=nqubits) # Test qibo - qibo.set_backend(backend=config.qibo['backend'], \ - platform=config.qibo['platform']) + qibo.set_backend(backend=config.qibo['backend'], + platform=config.qibo['platform']) start_time = timer() - qibo_circ, result_sv = qibo_qft(nqubits, init_state=init_state_qibo, \ - swaps=config.qibo['swaps']) + qibo_circ, result_sv = qibo_qft(nqubits, init_state=init_state_qibo, + swaps=config.qibo['swaps']) end_time = timer() qibo_time = end_time - start_time @@ -45,9 +46,10 @@ def test_eval(nqubits: int): # Test quimb start_time = timer() - result_tn = qasm_quimb.eval_QI_qft(nqubits=nqubits, qasm_circ=qasm_circ, \ - init_state=init_state_for_tn, backend=config.quimb['backend'], \ - swaps=config.quimb['swaps']) + result_tn = qasm_quimb.eval_QI_qft(nqubits=nqubits, qasm_circ=qasm_circ, + init_state=init_state_for_tn, + backend=config.quimb['backend'], + swaps=config.quimb['swaps']) end_time = timer() quimb_time = end_time - start_time