update example 5

amazon-braket · Nov 28, 2023 · 55e2e7b · 55e2e7b
1 parent 448fda4
commit 55e2e7b
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/examples/autoqasm/5_Pulse_programming_and_dynamical_decoupling.ipynb b/examples/autoqasm/5_Pulse_programming_and_dynamical_decoupling.ipynb
@@ -37,7 +37,7 @@
     "\n",
     "import braket.experimental.autoqasm as aq\n",
     "from braket.experimental.autoqasm import pulse\n",
-    "from braket.experimental.autoqasm.instructions import rx, rz, measure\n",
+    "from braket.experimental.autoqasm.instructions import rx, rz\n",
     "\n",
     "from braket.aws import AwsDevice\n",
     "from braket.devices import Devices\n",
@@ -69,8 +69,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "# TODO: determine if this example should make these variables inputs to the program\n",
+    "qubit = 0\n",
+    "idle_duration = 10e-6\n",
+    "\n",
     "@aq.main\n",
-    "def idle(qubit: int, idle_duration: float):\n",
+    "def idle():\n",
     "    control_frame = device.frames[f\"q{qubit}_rf_frame\"]\n",
     "    pulse.delay(control_frame, idle_duration)\n",
     "    pulse.capture_v0(control_frame)"
@@ -104,7 +108,7 @@
     }
    ],
    "source": [
-    "print(idle(0, 10e-6).to_ir())"
+    "print(idle.to_ir())"
    ]
   },
   {
@@ -180,8 +184,12 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "qubit = 0\n",
+    "idle_duration = 10e-6\n",
+    "n_cycles = 3\n",
+    "\n",
     "@aq.main\n",
-    "def idle_with_dd(qubit: int, idle_duration: float, n_cycles: int = 1) -> None:\n",
+    "def idle_with_dd():\n",
     "    dd_spacing = idle_duration / (4*n_cycles)\n",
     "\n",
     "    control_frame = device.frames[f\"q{qubit}_rf_frame\"]\n",
@@ -247,7 +255,7 @@
     }
    ],
    "source": [
-    "print(idle_with_dd(0, 10e-6, 3).to_ir())"
+    "print(idle_with_dd.to_ir())"
    ]
   },
   {