Use autopep8 for PEP-8 formatting

This commit is contained in:
Liwei Yang
2023-02-15 17:28:06 +08:00
parent bc13eed8f4
commit 53382b8184
2 changed files with 24 additions and 16 deletions

View File

@@ -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
return amplitudes