restore append in overlapo with density mode and added pragma

qiboteam · Aug 10, 2023 · 219db62 · 219db62
1 parent 503a2d0
commit 219db62
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/qibo/callbacks.py b/src/qibo/callbacks.py
@@ -206,8 +206,10 @@ def apply(self, backend, state):
         self.append(overlap)
         return overlap
 
-    def apply_density_matrix(self, backend, state):
-        return backend.calculate_overlap_density_matrix(self.state, state)
+    def apply_density_matrix(self, backend, state):  # pragma: no cover
+        overlap = backend.calculate_overlap_density_matrix(self.state, state)
+        self.append(overlap)
+        return overlap
 
 
 class Energy(Callback):