Skip to content

Commit

Permalink
fix mypy
Browse files Browse the repository at this point in the history
  • Loading branch information
kevinsung committed Aug 20, 2024
1 parent 4c40f28 commit 3558ea3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/ffsim/linalg/double_factorized_decomposition.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,7 +34,7 @@ def _truncated_eigh(
indices = np.argsort(np.abs(eigs))[::-1]
eigs = eigs[indices]
vecs = vecs[:, indices]
n_discard = np.searchsorted(np.cumsum(np.abs(eigs[::-1])), tol)
n_discard = int(np.searchsorted(np.cumsum(np.abs(eigs[::-1])), tol))
n_vecs = cast(int, min(max_vecs, len(eigs) - n_discard))
return eigs[:n_vecs], vecs[:, :n_vecs]

Expand Down

0 comments on commit 3558ea3

Please sign in to comment.