fixed a bug

Author: orsa-classiq

algorithms/hamiltonian_simulation/hamiltonian_simulation_with_block_encoding/hamiltonian_simulation_with_block_encoding.ipynb

@@ -473,11 +473,17 @@
     "\n",
     "\n",
     "@qfunc\n",
+    "def my_reflect_about_zero(qba: QNum):\n",
+    "    control(qba == 0, lambda: phase(pi))\n",
+    "    phase(pi)\n",
+    "\n",
+    "\n",
+    "@qfunc\n",
     "def walk_operator(\n",
     "    be_qfunc: QCallable[BlockEncodedState], state: BlockEncodedState\n",
     ") -> None:\n",
     "    be_qfunc(state)\n",
-    "    control(state.block == 0, lambda: phase(pi))"
+    "    my_reflect_about_zero(state.block)"
    ]
   },
   {

algorithms/quantum_phase_estimation/qpe_with_qubitization/qpe_for_molecule_with_qubitization.ipynb

@@ -359,11 +359,17 @@
     "\n",
     "\n",
     "@qfunc\n",
+    "def my_reflect_about_zero(qba: QNum):\n",
+    "    control(qba == 0, lambda: phase(pi))\n",
+    "    phase(pi)\n",
+    "\n",
+    "\n",
+    "@qfunc\n",
     "def walk_operator(\n",
     "    be_qfunc: QCallable[BlockEncodedState], state: BlockEncodedState\n",
     ") -> None:\n",
     "    be_qfunc(state)\n",
-    "    control(state.block == 0, lambda: phase(pi))"
+    "    my_reflect_about_zero(state.block)"
    ]
   },
   {

applications/cfd/qls_for_hybrid_solvers/qls_chebyshev_lcu.ipynb

@@ -50,7 +50,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": 4,
    "id": "3",
    "metadata": {},
    "outputs": [],
@@ -59,8 +59,9 @@
     "\n",
     "\n",
     "@qfunc\n",
-    "def my_reflect_about_zero(qba: QArray):\n",
+    "def my_reflect_about_zero(qba: QNum):\n",
     "    control(qba == 0, lambda: phase(pi))\n",
+    "    phase(pi)\n",
     "\n",
     "\n",
     "@qfunc\n",

community/paper_implementation_project/quantum_state_preparation_without_coherent_arithmetic/stateprep_guassian_using_qsvt.ipynb

@@ -78,7 +78,7 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "0.98.0\n"
+      "1.1.0\n"
      ]
     }
    ],
@@ -454,25 +454,32 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 10,
+   "execution_count": 15,
    "id": "25",
    "metadata": {},
    "outputs": [],
    "source": [
+    "from classiq.qmod.symbolic import pi\n",
+    "\n",
+    "\n",
     "@qfunc\n",
     "def state_prep(reg: MyQStruct):\n",
     "    hadamard_transform(reg.x)\n",
     "    u_f(reg)\n",
     "\n",
     "\n",
     "@qfunc\n",
+    "def mark_zero_state(q: QNum):\n",
+    "    control(q == 0, lambda: phase(pi))\n",
+    "\n",
+    "\n",
+    "@qfunc\n",
     "def u_amp(reg: MyQStruct):\n",
     "    \"\"\"Amplitude amplification circuit for state preparation using QET (more generally, QSVT)\"\"\"\n",
     "    exact_amplitude_amplification(\n",
     "        amplitude=amp,\n",
-    "        oracle=lambda r: reflect_about_zero(\n",
-    "            r[reg.x.size : r.len]\n",
-    "        ),  # Mark if the state is good state. Auxiliary qubits are |0> => good state\n",
+    "        # Mark if the state is good state. Auxiliary qubits are |0> => good state\n",
+    "        oracle=lambda r: mark_zero_state(r[reg.x.size : r.len]),\n",
     "        space_transform=lambda _reg: state_prep(_reg),\n",
     "        packed_qvars=[reg.x, reg.a1, reg.a2, reg.a3],\n",
     "    )"
@@ -496,15 +503,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 11,
+   "execution_count": 16,
    "id": "28",
    "metadata": {},
    "outputs": [
     {
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Quantum program link: https://platform.classiq.io/circuit/35YpmHOXeOGaZFnfA6v7HWDqmsS\n"
+      "Quantum program link: https://platform.classiq.io/circuit/3A6ojZBlJr4z9V2zIN2pZCip4xE\n"
      ]
     }
    ],
@@ -540,7 +547,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 12,
+   "execution_count": 17,
    "id": "29",
    "metadata": {},
    "outputs": [
@@ -558,8 +565,8 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "Mean Squared Error (MSE): 2.2611239359570596e-11\n",
-      "{'depth': 7511, 'width': 14, 'cx_count': 4073}\n"
+      "Mean Squared Error (MSE): 2.2611239362935673e-11\n",
+      "{'depth': 8000, 'width': 12, 'cx_count': 4573}\n"
      ]
     }
    ],