fix check for whether dtype is complex

qiskit-community · Sep 20, 2023 · cb2cc9d · cb2cc9d
1 parent e799a70
commit cb2cc9d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ffsim/random.py b/python/ffsim/random.py
@@ -27,7 +27,7 @@ def random_statevector(dim: int, *, seed=None, dtype=complex) -> np.ndarray:
     """
     rng = np.random.default_rng(seed)
     vec = rng.standard_normal(dim).astype(dtype, copy=False)
-    if np.iscomplexobj(dtype):
+    if np.issubdtype(dtype, np.complexfloating):
         vec += 1j * rng.standard_normal(dim).astype(dtype, copy=False)
     vec /= np.linalg.norm(vec)
     return vec