ES changed, now computing fid and entropy

mgbukov · Sep 17, 2017 · c7723bf · c7723bf
1 parent 1c32787
commit c7723bf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/SA_v2/main.py b/SA_v2/main.py
@@ -580,15 +580,15 @@ def run_ES(parameters, model:MODEL, utils):
     psi = model.compute_evolved_state()
     model.compute_fidelity(psi_evolve=psi)
     model.compute_Sent(psi_evolve=psi)
-    #model.compute_energy(psi_evolve=psi)
+    model.compute_energy(psi_evolve=psi)
     print("Est. run time : \t %.3f s"%(0.5*n_protocol*(time.time()-st)))
     # ---> Starting real calculation <---
 
     st=time.time()
     for p in range(n_protocol):
         model.update_protocol(b2_array(p))
         psi = model.compute_evolved_state()
-        exact_data[p] = (model.compute_fidelity(psi_evolve=psi), model.compute_Sent(psi_evolve=psi))
+        exact_data[p] = (model.compute_fidelity(psi_evolve=psi), model.compute_energy(psi_evolve=psi), model.compute_Sent(psi_evolve=psi))
 
     outfile = utils.make_file_name(parameters, root=parameters['root'])
     with open(outfile,'wb') as f: