add more bloq examples

quantumlib · Mar 22, 2024 · 0d08279 · 0d08279
1 parent db1a412
commit 0d08279
Show file tree

Hide file tree

Showing 2 changed files with 39 additions and 3 deletions.
diff --git a/qualtran/bloqs/basic_gates/su2_rotation.ipynb b/qualtran/bloqs/basic_gates/su2_rotation.ipynb
@@ -86,6 +86,30 @@
     "su2_rotation_gate = SU2RotationGate(np.pi / 4, np.pi / 2, np.pi / 2)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "7233599d",
+   "metadata": {
+    "cq.autogen": "SU2RotationGate.hadamard"
+   },
+   "outputs": [],
+   "source": [
+    "hadamard = SU2RotationGate(np.pi / 4, 0, 0)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9f659e0d",
+   "metadata": {
+    "cq.autogen": "SU2RotationGate.t_gate"
+   },
+   "outputs": [],
+   "source": [
+    "t_gate = SU2RotationGate(0, 3 * np.pi / 4, 0)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "9a2820b5",
@@ -106,8 +130,8 @@
    "outputs": [],
    "source": [
     "from qualtran.drawing import show_bloqs\n",
-    "show_bloqs([su2_rotation_gate],\n",
-    "           ['`su2_rotation_gate`'])"
+    "show_bloqs([su2_rotation_gate, hadamard, t_gate],\n",
+    "           ['`su2_rotation_gate`', '`hadamard`', '`t_gate`'])"
    ]
   },
   {

diff --git a/qualtran/bloqs/basic_gates/su2_rotation.py b/qualtran/bloqs/basic_gates/su2_rotation.py
@@ -105,8 +105,20 @@ def _su2_rotation_gate() -> SU2RotationGate:
     return su2_rotation_gate
 
 
+@bloq_example
+def _hadamard() -> SU2RotationGate:
+    hadamard = SU2RotationGate(np.pi / 4, 0, 0)
+    return hadamard
+
+
+@bloq_example
+def _t_gate() -> SU2RotationGate:
+    t_gate = SU2RotationGate(0, 3 * np.pi / 4, 0)
+    return t_gate
+
+
 _SU2_ROTATION_GATE_DOC = BloqDocSpec(
     bloq_cls=SU2RotationGate,
     import_line='from qualtran.bloqs.basic_gates import SU2RotationGate',
-    examples=[_su2_rotation_gate],
+    examples=[_su2_rotation_gate, _hadamard, _t_gate],
 )