doc: correct gphase matrix representation (#946)

amazon-braket · Apr 12, 2024 · b1cd32d · b1cd32d
1 parent 53aba5e
commit b1cd32d
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/braket/circuits/gates.py b/src/braket/circuits/gates.py
@@ -220,9 +220,8 @@ class GPhase(AngledGate):
 
     Unitary matrix:
 
-        .. math:: \mathtt{gphase}(\gamma) = e^{i \gamma} I = \begin{bmatrix}
-                    e^{i \gamma} & 0 \\
-                    0 & e^{i \gamma} \end{bmatrix}.
+        .. math:: \mathtt{gphase}(\gamma) = e^{i \gamma} I_1 = \begin{bmatrix}
+                    e^{i \gamma} \end{bmatrix}.
 
     Args:
         angle (Union[FreeParameterExpression, float]): angle in radians.
@@ -279,9 +278,8 @@ def gphase(
 
         Unitary matrix:
 
-            .. math:: \mathtt{gphase}(\gamma) = e^{i \gamma} I = \begin{bmatrix}
-                        e^{i \gamma} & 0 \\
-                        0 & e^{i \gamma} \end{bmatrix}.
+            .. math:: \mathtt{gphase}(\gamma) = e^{i \gamma} I_1 = \begin{bmatrix}
+                        e^{i \gamma} \end{bmatrix}.
 
         Args:
             angle (Union[FreeParameterExpression, float]): Phase in radians.