state_props:eof

toqito/state_props/entanglement_of_formation.py

@@ -1,5 +1,6 @@
 """Computes the entanglement of formation of a bipartite quantum state."""
 
+from __future__ import annotations
 
 import numpy as np
 import scipy
@@ -65,7 +66,7 @@ def entanglement_of_formation(rho: np.ndarray, dim: list[int] | int = None) -> f
 
     # User can specify dimension as integer.
     if isinstance(dim, int):
-        dim = np.array([dim, max(dim_x, dim_y) / dim], dtype=int)
+        dim = np.array([dim, max(dim_x, dim_y) / dim], dtype=int) # pylint: disable=redefined-variable-type
         if abs(dim[1] - np.round(dim[1])) >= 2 * max(dim_x, dim_y) * eps:
             raise ValueError(
                 "Invalid dimension: If `dim` is provided as a "
@@ -87,7 +88,7 @@ def entanglement_of_formation(rho: np.ndarray, dim: list[int] | int = None) -> f
     # Start computing entanglement-of-formation.
     if min(dim_x, dim_y) == 1:
         rho = rho[:]
-        dim = [x for x in dim]
+        dim = [int(x) for x in dim]
         return von_neumann_entropy(partial_trace(rho * rho.conj().T, [1], dim))
 
     # Case: :code:`rho` is a density matrix.