Add S and T gates to circuit.draw()

qiboteam · Jan 20, 2022 · 492807d · 492807d
1 parent 86fa8b7
commit 492807d
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/qibo/abstractions/circuit.py b/src/qibo/abstractions/circuit.py
@@ -159,7 +159,7 @@ def on_qubits(self, *q):
             q (int): Qubit ids that the gates should act.
 
         Example:
-            
+
             .. testcode::
 
                 from qibo import gates, models
@@ -664,7 +664,7 @@ def summary(self) -> str:
                 cx: 2
                 ccx: 1
                 '''
-        
+
             .. testoutput::
                 :hide:
 
@@ -767,7 +767,7 @@ def from_qasm(cls, qasm_code: str, **kwargs):
             specified by the given QASM script.
 
         Example:
-         
+
             .. testcode::
 
                 from qibo import models, gates
@@ -948,6 +948,7 @@ def draw(self, line_wrap=70) -> str:
             String containing text circuit diagram.
         """
         labels = {"h": "H", "x": "X", "y": "Y", "z": "Z",
+                  "s": "S", "sdg": "SDG", "t": "T", "tdg": "TDG",
                   "rx": "RX", "ry": "RY", "rz": "RZ",
                   "u1": "U1", "u2": "U2", "u3": "U3",
                   "cx": "X", "swap": "x", "cz": "Z",