fix slater_determinant_rdms overload variants

qiskit-community · Jun 28, 2024 · 3e924cd · 3e924cd
1 parent bbe42fa
commit 3e924cd
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/ffsim/states/states.py b/python/ffsim/states/states.py
@@ -276,6 +276,8 @@ def slater_determinant_rdms(
     norb: int,
     occupied_orbitals: Sequence[int],
     orbital_rotation: np.ndarray | None = None,
+    *,
+    rank: int = 1,
 ) -> np.ndarray: ...
 @overload
 def slater_determinant_rdms(
@@ -284,6 +286,8 @@ def slater_determinant_rdms(
     orbital_rotation: np.ndarray
     | tuple[np.ndarray | None, np.ndarray | None]
     | None = None,
+    *,
+    rank: int = 1,
 ) -> np.ndarray: ...
 def slater_determinant_rdms(
     norb: int,