fix implementation of Rz

derwind · Mar 1, 2024 · e86d9b1 · e86d9b1
1 parent 2422b5b
commit e86d9b1
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ttz/mps/mps.py b/ttz/mps/mps.py
@@ -232,7 +232,10 @@ def Ry(theta: float):
 
 def Rz(theta: float):
     return np.array(
-        [[np.cos(theta / 2) - np.sin(theta / 2), 0], [0, np.cos(theta / 2) + np.sin(theta / 2)]],
+        [
+            [np.cos(theta / 2) - np.sin(theta / 2) * 1j, 0],
+            [0, np.cos(theta / 2) + np.sin(theta / 2) * 1j],
+        ],
         dtype=complex,
     )