fix docstr hamevo

pasqal-io · Nov 25, 2024 · aa4e12a · aa4e12a
1 parent 40d32f5
commit aa4e12a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/qadence/operations/ham_evo.py b/qadence/operations/ham_evo.py
@@ -64,6 +64,7 @@ class HamEvo(TimeEvolutionBlock):
     from qadence import X, HamEvo, PI, add, run
     from qadence import FeatureParameter, TimeParameter
     import torch
+    from pyqtorch.utils import SolverType
 
     n_qubits = 3
 
@@ -86,7 +87,8 @@ class HamEvo(TimeEvolutionBlock):
     # Adding noise operators
     noise_ops = [torch.eye(2**n_qubits, dtype=torch.complex128)]
     hevo = HamEvo(hamiltonian, parameter=t, noise_operators=noise_ops)
-    state = run(hevo, values = {"duration": torch.tensor(1.0)})
+    config = {"ode_solver": SolverType.DP5_ME, "n_steps_hevo": 100}
+    densitymat = run(hevo, values = {"duration": torch.tensor(1.0)})
     ```
     """