Skip to content

Commit

Permalink
Fix RB plot bug (#18)
Browse files Browse the repository at this point in the history 
* fixes to plot function in randomized_benchmarking_common.py so that individual decays are indexed correctly

* updated CHANGELOG.rst

* updated CHANGELOG.rst

---------

Co-authored-by: Pedro Figueroa <pedro.romero@meetiqm.com>
  • Loading branch information
@pedrofigro
pedrofigro and Pedro Figueroa authored Dec 16, 2024
1 parent 0102c27 commit b177630
Show file tree
Hide file tree
Showing 2 changed files with 47 additions and 24 deletions.
4 changes: 4 additions & 0 deletions CHANGELOG.rst
View file
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,10 @@
Changelog
=========

Version 2.1
===========
* Fixed bug in RB plots for individual decays.

Version 2.0
===========
* Adds `Circuits`, `BenchmarkCircuit` and `CircuitGroup` as a way to easily store and interact with multiple quantum circuits.
Expand Down
67 changes: 43 additions & 24 deletions src/iqm/benchmarks/randomized_benchmarking/randomized_benchmarking_common.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -565,6 +565,11 @@ def plot_rb_decay(
num_circuit_samples = dataset.attrs["num_circuit_samples"]
timestamp = dataset.attrs["execution_timestamp"]
backend_name = dataset.attrs["backend_name"]
# If only one layout is passed, the index to retrieve results must be shifted!
qubits_index = 0
if len(qubits_array) == 1:
config_qubits_array = dataset.attrs["qubits_array"]
qubits_index = config_qubits_array.index(qubits_array[0])

# Fetch the relevant observations indexed by qubit layouts
depths = {}
Expand All @@ -583,79 +588,93 @@ def plot_rb_decay(
colors = [cmap(i) for i in np.linspace(start=1, stop=0, num=len(qubits_array)).tolist()]
if identifier == "mrb":
depths[identifier] = {
str(q): list(dataset.attrs[q_idx]["polarizations"].keys()) for q_idx, q in enumerate(qubits_array)
str(q): list(dataset.attrs[q_idx]["polarizations"].keys())
for q_idx, q in enumerate(qubits_array, qubits_index)
}
polarizations[identifier] = {
str(q): dataset.attrs[q_idx]["polarizations"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["polarizations"] for q_idx, q in enumerate(qubits_array, qubits_index)
}
average_polarizations[identifier] = {
str(q): dataset.attrs[q_idx]["avg_polarization_nominal_values"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["avg_polarization_nominal_values"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
stddevs_from_mean[identifier] = {
str(q): dataset.attrs[q_idx]["avg_polatization_stderr"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["avg_polatization_stderr"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
else:
else: # identifier == "clifford"
depths[identifier] = {
str(q): list(dataset.attrs[q_idx]["fidelities"].keys()) for q_idx, q in enumerate(qubits_array)
str(q): list(dataset.attrs[q_idx]["fidelities"].keys())
for q_idx, q in enumerate(qubits_array, qubits_index)
}
polarizations[identifier] = {
str(q): dataset.attrs[q_idx]["fidelities"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["fidelities"] for q_idx, q in enumerate(qubits_array, qubits_index)
}
average_polarizations[identifier] = {
str(q): dataset.attrs[q_idx]["avg_fidelities_nominal_values"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["avg_fidelities_nominal_values"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
stddevs_from_mean[identifier] = {
str(q): dataset.attrs[q_idx]["avg_fidelities_stderr"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["avg_fidelities_stderr"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
# These are common to both MRB and standard Clifford
fidelity_value[identifier] = {
str(q): observations[q_idx]["avg_gate_fidelity"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): observations[q_idx]["avg_gate_fidelity"]["value"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
fidelity_stderr[identifier] = {
str(q): observations[q_idx]["avg_gate_fidelity"]["uncertainty"] for q_idx, q in enumerate(qubits_array)
str(q): observations[q_idx]["avg_gate_fidelity"]["uncertainty"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
decay_rate[identifier] = {
str(q): dataset.attrs[q_idx]["decay_rate"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["decay_rate"]["value"] for q_idx, q in enumerate(qubits_array, qubits_index)
}
offset[identifier] = {
str(q): dataset.attrs[q_idx]["fit_offset"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["fit_offset"]["value"] for q_idx, q in enumerate(qubits_array, qubits_index)
}
amplitude[identifier] = {
str(q): dataset.attrs[q_idx]["fit_amplitude"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx]["fit_amplitude"]["value"] for q_idx, q in enumerate(qubits_array, qubits_index)
}
else:
else: # id MRB
rb_type_keys = list(observations[0].keys())
colors = [cmap(i) for i in np.linspace(start=1, stop=0, num=len(rb_type_keys)).tolist()]
for rb_type in rb_type_keys:
depths[rb_type] = {
str(q): list(dataset.attrs[q_idx][rb_type]["fidelities"].keys()) for q_idx, q in enumerate(qubits_array)
str(q): list(dataset.attrs[q_idx][rb_type]["fidelities"].keys())
for q_idx, q in enumerate(qubits_array, qubits_index)
}
polarizations[rb_type] = {
str(q): dataset.attrs[q_idx][rb_type]["fidelities"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx][rb_type]["fidelities"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
average_polarizations[rb_type] = {
str(q): dataset.attrs[q_idx][rb_type]["avg_fidelities_nominal_values"]
for q_idx, q in enumerate(qubits_array)
for q_idx, q in enumerate(qubits_array, qubits_index)
}
stddevs_from_mean[rb_type] = {
str(q): dataset.attrs[q_idx][rb_type]["avg_fidelities_stderr"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx][rb_type]["avg_fidelities_stderr"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
fidelity_value[rb_type] = {
str(q): observations[q_idx][rb_type]["avg_gate_fidelity"]["value"]
for q_idx, q in enumerate(qubits_array)
for q_idx, q in enumerate(qubits_array, qubits_index)
}
fidelity_stderr[rb_type] = {
str(q): observations[q_idx][rb_type]["avg_gate_fidelity"]["uncertainty"]
for q_idx, q in enumerate(qubits_array)
for q_idx, q in enumerate(qubits_array, qubits_index)
}
decay_rate[rb_type] = {
str(q): dataset.attrs[q_idx][rb_type]["decay_rate"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx][rb_type]["decay_rate"]["value"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
offset[rb_type] = {
str(q): dataset.attrs[q_idx][rb_type]["fit_offset"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx][rb_type]["fit_offset"]["value"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}
amplitude[rb_type] = {
str(q): dataset.attrs[q_idx][rb_type]["fit_amplitude"]["value"] for q_idx, q in enumerate(qubits_array)
str(q): dataset.attrs[q_idx][rb_type]["fit_amplitude"]["value"]
for q_idx, q in enumerate(qubits_array, qubits_index)
}

for index_irb, key in enumerate(rb_type_keys):
Expand Down

0 comments on commit b177630

Please sign in to comment.