Update tests/test_quantum_info_entropies.py

Co-authored-by: BrunoLiegiBastonLiegi <[email protected]>
qiboteam · Oct 3, 2024 · 9a9c9e5 · 9a9c9e5
1 parent d644294
commit 9a9c9e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_quantum_info_entropies.py b/tests/test_quantum_info_entropies.py
@@ -393,8 +393,8 @@ def test_classical_relative_tsallis_entropy(backend, alpha, base, kind):
     prob_dist_q = np.random.rand(10)
     prob_dist_q /= np.sum(prob_dist_q)
 
-    prob_dist_p = backend.np.real(backend.cast(prob_dist_p))
-    prob_dist_q = backend.np.real(backend.cast(prob_dist_q))
+    prob_dist_p = backend.cast(prob_dist_p, dtype=np.float64)
+    prob_dist_q = backend.cast(prob_dist_q, dtype=np.float64)
 
     if alpha == 1.0:
         target = classical_relative_entropy(prob_dist_p, prob_dist_q, base, backend)