[minor] adding imports for pulse

mit-han-lab · Nov 24, 2023 · 1637680 · 1637680
1 parent 83345b4
commit 1637680
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 5 deletions.
diff --git a/examples/optimal_control/optimal_control.py b/examples/optimal_control/optimal_control.py
@@ -41,7 +41,7 @@
         dtype=torch.complex64,
     )
 
-    pulse = tq.QuantumPulseDirect(n_steps=4, hamil=[[0, 1], [1, 0]])
+    pulse = tq.pulse.QuantumPulseDirect(n_steps=4, hamil=[[0, 1], [1, 0]])
 
     optimizer = optim.Adam(params=pulse.parameters(), lr=5e-3)
 

diff --git a/examples/optimal_control/optimal_control_gaussian.py b/examples/optimal_control/optimal_control_gaussian.py
@@ -41,7 +41,7 @@
         dtype=torch.complex64,
     )
 
-    pulse = tq.QuantumPulseGaussian(hamil=[[0, 1], [1, 0]])
+    pulse = tq.pulse.QuantumPulseGaussian(hamil=[[0, 1], [1, 0]])
 
     optimizer = optim.Adam(params=pulse.parameters(), lr=5e-3)
 

diff --git a/examples/optimal_control/optimal_control_multi_qubit.py b/examples/optimal_control/optimal_control_multi_qubit.py
@@ -43,9 +43,9 @@
         dtype=torch.complex64,
     )
 
-    pulse_q0 = tq.QuantumPulseDirect(n_steps=10, hamil=[[0, 1], [1, 0]])
-    pulse_q1 = tq.QuantumPulseDirect(n_steps=10, hamil=[[0, 1], [1, 0]])
-    pulse_q01 = tq.QuantumPulseDirect(
+    pulse_q0 = tq.pulse.QuantumPulseDirect(n_steps=10, hamil=[[0, 1], [1, 0]])
+    pulse_q1 = tq.pulse.QuantumPulseDirect(n_steps=10, hamil=[[0, 1], [1, 0]])
+    pulse_q01 = tq.pulse.QuantumPulseDirect(
         n_steps=10,
         hamil=[
             [1, 0, 0, 0],

diff --git a/torchquantum/__init__.py b/torchquantum/__init__.py
@@ -38,6 +38,7 @@
 from .noise_model import *
 from .algorithm import *
 from .dataset import *
+from .pulse import *
 
 # here we check whether the Qiskit parameterization bug is fixed, if not, a
 # warning message will be printed

diff --git a/torchquantum/pulse/__init__.py b/torchquantum/pulse/__init__.py
@@ -25,4 +25,5 @@
 from .utils import *
 from .sesolve import sesolve
 from .mesolve import mesolve
+from .pulses import *
 # from .smesolve import smesolve