fix: add symbol setter

chensgit169 · Dec 23, 2023 · 008eb33 · 008eb33
1 parent f8102e9
commit 008eb33
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/quafu/elements/quantum_gate.py b/quafu/elements/quantum_gate.py
@@ -30,9 +30,12 @@ def __init__(self,
                  ):
         super().__init__(pos, paras)
         self._matrix = matrix
+        self._symbol = None
 
     @property
     def symbol(self):
+        if self._symbol is not None:
+            return self._symbol
         if self.paras is not None:
             if isinstance(self.paras, Iterable):
                 symbol = "%s(" % self.name + ",".join(["%.3f" % para for para in self.paras]) + ")"
@@ -42,6 +45,10 @@ def symbol(self):
             symbol = "%s" % self.name
         return symbol
 
+    @symbol.setter
+    def symbol(self, symbol):
+        self._symbol = symbol
+
     def update_params(self, paras: Union[float, List[float]]):
         """Update parameters of this gate"""
         if paras is None: