Update quantum_pulse_simulation.py

mit-han-lab · Jun 12, 2024 · 82cf184 · 82cf184
1 parent 0f0de3f
commit 82cf184
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torchquantum/pulse/quantum_pulse_simulation.py b/torchquantum/pulse/quantum_pulse_simulation.py
@@ -14,6 +14,11 @@ class QuantumPulseDemo(tq.QuantumModule):
     def __init__(self):
         """
         Initializes the QuantumPulseDemo module.
+
+        Args:
+            n_wires (int): Number of quantum wires (qubits).
+            n_steps (int): Number of steps for the quantum pulse.
+            hamil (list): Hamiltonian for the quantum pulse.
         """
         super().__init__()
         self.n_wires = 2