Skip to content

Commit

Permalink
Use int64 instead of int_
Browse files Browse the repository at this point in the history
  • Loading branch information
sidjha1 committed Nov 3, 2024
1 parent fc67fa7 commit eeff8bf
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion lotus/sem_ops/cascade_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
def importance_sampling(
proxy_scores: list[float],
sample_percentage: float,
) -> tuple[NDArray[np.int_], NDArray[np.float64]]:
) -> tuple[NDArray[np.int64], NDArray[np.float64]]:
"""Uses importance sampling and returns the list of indices from which to learn cascade thresholds."""

w = np.sqrt(proxy_scores)
Expand Down

0 comments on commit eeff8bf

Please sign in to comment.