Update graphix/pauli.py

Co-authored-by: S.S. <66886825+EarlMilktea@users.noreply.github.com>
jemappellesami · Aug 21, 2024 · df17ca3 · df17ca3
1 parent d4a6073
commit df17ca3
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/graphix/pauli.py b/graphix/pauli.py
@@ -263,13 +263,14 @@ def get_eigenstate(self, eigenvalue=0) -> BasicStates:
 
         if self.symbol == IXYZ.X:
             return BasicStates.PLUS if eigenvalue == 0 else BasicStates.MINUS
-        elif self.symbol == IXYZ.Y:
+        if self.symbol == IXYZ.Y:
             return BasicStates.PLUS_I if eigenvalue == 0 else BasicStates.MINUS_I
-        elif self.symbol == IXYZ.Z:
+        if self.symbol == IXYZ.Z:
             return BasicStates.ZERO if eigenvalue == 0 else BasicStates.ONE
         # Any state is eigenstate of the identity
-        elif self.symbol == IXYZ.I:
+        if self.symbol == IXYZ.I:
             return BasicStates.PLUS
+        typing_extensions.assert_never(self.symbol)
 
     def __repr__(self) -> str:
         return self.__unit.prefix(self.__symbol.name)