change clip to where

rochisha0 · Aug 8, 2024 · 8995615 · 8995615
1 parent 234f26b
commit 8995615
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/qutip/entropy.py b/qutip/entropy.py
@@ -37,7 +37,8 @@ def entropy_vn(rho, base=np.e, sparse=False):
     if rho.type == 'ket' or rho.type == 'bra':
         rho = ket2dm(rho)
     vals = rho.eigenenergies(sparse=sparse)
-    nzvals = np.clip(vals, 1e-17, None)
+    threshold = 1e-17
+    nzvals = np.where(vals < threshold, threshold, vals)
     if base == 2:
         logvals = np.log2(nzvals)
     elif base == np.e: