preserve type

qiskit-community · Nov 22, 2024 · db32932 · db32932
1 parent 60c8aa7
commit db32932
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ffsim/variational/util.py b/python/ffsim/variational/util.py
@@ -120,7 +120,7 @@ def orbital_rotation_from_t1_amplitudes(t1: np.ndarray) -> np.ndarray:
     """
     nocc, nvrt = t1.shape
     norb = nocc + nvrt
-    generator = np.zeros((norb, norb), dtype=complex)
+    generator = np.zeros((norb, norb), dtype=t1.dtype)
     generator[:nocc, nocc:] = t1
     generator[nocc:, :nocc] = -t1.T
     return scipy.linalg.expm(generator)