doc: Improve gphase unitary matrix definition in docstring (#944)

* Improve gphase unitary matrix definition in docstring * Remove extra whitespace --------- Co-authored-by: Abe Coull <85974725+math411@users.noreply.github.com>
amazon-braket · Apr 9, 2024 · c687223 · c687223
1 parent 64a2536
commit c687223
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/braket/circuits/gates.py b/src/braket/circuits/gates.py
@@ -220,7 +220,9 @@ class GPhase(AngledGate):
 
     Unitary matrix:
 
-        .. math:: \mathtt{gphase}(\gamma) = e^(i \gamma) I_1.
+        .. math:: \mathtt{gphase}(\gamma) = e^{i \gamma} I = \begin{bmatrix}
+                    e^{i \gamma} & 0 \\
+                    0 & e^{i \gamma} \end{bmatrix}.
 
     Args:
         angle (Union[FreeParameterExpression, float]): angle in radians.
@@ -277,7 +279,9 @@ def gphase(
 
         Unitary matrix:
 
-            .. math:: \mathtt{gphase}(\gamma) = e^(i \gamma) I_1.
+            .. math:: \mathtt{gphase}(\gamma) = e^{i \gamma} I = \begin{bmatrix}
+                        e^{i \gamma} & 0 \\
+                        0 & e^{i \gamma} \end{bmatrix}.
 
         Args:
             angle (Union[FreeParameterExpression, float]): Phase in radians.