fix: fix type annotations

mit-han-lab · Jun 12, 2024 · cd2f3d4 · cd2f3d4
1 parent 64919c2
commit cd2f3d4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/torchquantum/plugin/qiskit/qiskit_plugin.py b/torchquantum/plugin/qiskit/qiskit_plugin.py
@@ -21,8 +21,9 @@
 OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 SOFTWARE.
 """
+from __future__ import annotations
 
-from typing import Iterable
+from typing import Iterable, Optional
 
 import numpy as np
 import qiskit
@@ -837,7 +838,7 @@ def qiskit2tq_translate_qiskit_params(
 
 
 def qiskit2tq(
-    circ: QuantumCircuit, initial_parameters: list[torch.nn.Parameter] | None = None
+    circ: QuantumCircuit, initial_parameters: Optional[list[torch.nn.Parameter]] = None
 ):
     ops = qiskit2tq_Operator(circ, initial_parameters)
     return tq.QuantumModuleFromOps(ops)