Skip to content

Commit f478067

Browse files
fix: update how to run local detuning AHS programs with the experimental capabilities context manager
1 parent 04c53e2 commit f478067

File tree

2 files changed

+13
-9
lines changed

2 files changed

+13
-9
lines changed

examples/analog_hamiltonian_simulation/07_Simulating_Lattice_Gauge_Theory_with_Rydberg_Atoms.ipynb

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
},
1111
"source": [
1212
"<div class=\"alert alert-block alert-info\">\n",
13-
"<b>Note:</b> This notebook requires you to have enabled Experimental Capabilities through Braket Direct. Follow these <a href='https://docs.aws.amazon.com/braket/latest/developerguide/braket-experimental-capabilities.html'>steps</a> to enable access to local detuning.\n",
13+
"<b>Note:</b> This notebook we use Experimental Capabilities. Read more about it <a href='https://docs.aws.amazon.com/braket/latest/developerguide/braket-experimental-capabilities.html'>here</a>.\n",
1414
"</div>"
1515
]
1616
},
@@ -78,6 +78,7 @@
7878
" show_final_avg_density,\n",
7979
")\n",
8080
"\n",
81+
"from braket.experimental_capabilities import EnableExperimentalCapability\n",
8182
"from braket.ahs.analog_hamiltonian_simulation import AnalogHamiltonianSimulation\n",
8283
"from braket.ahs.atom_arrangement import AtomArrangement\n",
8384
"from braket.ahs.driving_field import DrivingField\n",
@@ -443,10 +444,11 @@
443444
"outputs": [],
444445
"source": [
445446
"task_arns = []\n",
446-
"for ind, program in enumerate(programs):\n",
447-
" task = qpu.run(program.discretize(qpu), shots=100)\n",
448-
" task_arn = task.metadata()[\"quantumTaskArn\"]\n",
449-
" task_arns.append(task_arn)"
447+
"with EnableExperimentalCapability():\n",
448+
" for ind, program in enumerate(programs):\n",
449+
" task = qpu.run(program.discretize(qpu), shots=100)\n",
450+
" task_arn = task.metadata()[\"quantumTaskArn\"]\n",
451+
" task_arns.append(task_arn)"
450452
]
451453
},
452454
{
@@ -564,7 +566,7 @@
564566
"name": "python",
565567
"nbconvert_exporter": "python",
566568
"pygments_lexer": "ipython3",
567-
"version": "3.11.4"
569+
"version": "3.14.3"
568570
},
569571
"varInspector": {
570572
"cols": {

examples/analog_hamiltonian_simulation/08_Maximum_Weight_Independent_Set.ipynb

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
},
1111
"source": [
1212
"<div class=\"alert alert-block alert-info\">\n",
13-
"<b>Note:</b> This notebook requires you to have enabled Experimental Capabilities through Braket Direct. Follow these <a href='https://docs.aws.amazon.com/braket/latest/developerguide/braket-experimental-capabilities.html'>steps</a> to enable access to local detuning.\n",
13+
"<b>Note:</b> This notebook we use Experimental Capabilities. Read more about it <a href='https://docs.aws.amazon.com/braket/latest/developerguide/braket-experimental-capabilities.html'>here</a>.\n",
1414
"</div>"
1515
]
1616
},
@@ -79,6 +79,7 @@
7979
"import numpy as np\n",
8080
"from ahs_utils import show_drive_and_local_detuning, show_final_avg_density, show_register\n",
8181
"\n",
82+
"from braket.experimental_capabilities import EnableExperimentalCapability\n",
8283
"from braket.ahs.analog_hamiltonian_simulation import AnalogHamiltonianSimulation\n",
8384
"from braket.ahs.atom_arrangement import AtomArrangement\n",
8485
"from braket.ahs.driving_field import DrivingField\n",
@@ -358,7 +359,8 @@
358359
}
359360
],
360361
"source": [
361-
"task = qpu.run(program.discretize(qpu), shots=1000)\n",
362+
"with EnableExperimentalCapability():\n",
363+
" task = qpu.run(program.discretize(qpu), shots=1000)\n",
362364
"metadata = task.metadata()\n",
363365
"task_arn = metadata[\"quantumTaskArn\"]\n",
364366
"print(f\"ARN: {task_arn}\")"
@@ -486,7 +488,7 @@
486488
"name": "python",
487489
"nbconvert_exporter": "python",
488490
"pygments_lexer": "ipython3",
489-
"version": "3.11.4"
491+
"version": "3.14.3"
490492
},
491493
"varInspector": {
492494
"cols": {

0 commit comments

Comments
 (0)