remove hardcoded waveform names

amazon-braket · Aug 26, 2024 · 51d8aed · 51d8aed
1 parent dbf828c
commit 51d8aed
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 6 deletions.
diff --git a/examples/pulse_control/2_Native_gate_calibrations.ipynb b/examples/pulse_control/2_Native_gate_calibrations.ipynb
@@ -512,12 +512,14 @@
     }
    ],
    "source": [
+    "waveform_name = list(pulse_sequence_rx_pi_2_q30._waveforms.keys())[0]\n",
+    "\n",
     "lambda_ = FreeParameter(\"lambda_\")\n",
     "theta = FreeParameter(\"theta\")\n",
     "phi = FreeParameter(\"phi\")\n",
     "\n",
     "drive_frame = device.frames[\"Transmon_30_charge_tx\"]\n",
-    "x90 = pulse_sequence_rx_pi_2_q30._waveforms[\"wf_drag_gaussian_216\"]\n",
+    "x90 = pulse_sequence_rx_pi_2_q30._waveforms[waveform_name]\n",
     "\n",
     "U_sequence = (\n",
     "    PulseSequence()\n",

diff --git a/examples/pulse_control/3_Bell_pair_with_pulses_Rigetti.ipynb b/examples/pulse_control/3_Bell_pair_with_pulses_Rigetti.ipynb
@@ -309,8 +309,10 @@
     }
    ],
    "source": [
-    "a_b_cz_wfm = a_b_cz_wfms['wf_erf_square_847']\n",
-    "a_b_cz_frame = device.frames[f'Transmon_144_flux_tx_b']\n",
+    "waveform_name = list(a_b_cz._waveforms.keys())[0]\n",
+    "\n",
+    "a_b_cz_wfm = a_b_cz_wfms[waveform_name]\n",
+    "a_b_cz_frame = device.frames['Transmon_144_flux_tx_b']\n",
     "dt=a_b_cz_frame.port.dt\n",
     "\n",
     "print('CZ pulse duration:', round(a_b_cz_wfm.length * 1e9,0), 'ns')"

diff --git a/test/integ_tests/test_all_notebooks.py b/test/integ_tests/test_all_notebooks.py
@@ -21,9 +21,6 @@
     "04_Maximum_Independent_Sets_with_Analog_Hamiltonian_Simulation.ipynb",
     "05_Running_Analog_Hamiltonian_Simulation_with_local_simulator.ipynb",
     "09_Noisy_quantum_dynamics_for_Rydberg_atom_arrays.ipynb",
-    # Some pulse examples use hardcoded names that may not exist depending on the calibration data
-    "2_Native_gate_calibrations.ipynb",
-    "3_Bell_pair_with_pulses_Rigetti.ipynb",
     "Using_the_experimental_local_simulator.ipynb",
 ]