add ndarray type annotation to satisfy mypy

qiskit-community · Jan 2, 2025 · 16444ab · 16444ab
1 parent 92be4f1
commit 16444ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/ffsim/variational/ucj_spin_balanced.py b/python/ffsim/variational/ucj_spin_balanced.py
@@ -253,7 +253,7 @@ def from_parameters(
             pairs_aa = triu_indices
         if pairs_ab is None:
             pairs_ab = triu_indices
-        diag_coulomb_mats = np.zeros((n_reps, 2, norb, norb))
+        diag_coulomb_mats: np.ndarray = np.zeros((n_reps, 2, norb, norb))
         orbital_rotations = np.zeros((n_reps, norb, norb), dtype=complex)
         index = 0
         for orbital_rotation, diag_coulomb_mat in zip(