diff --git a/scipy-stubs/optimize/_lsap.pyi b/scipy-stubs/optimize/_lsap.pyi index 5cc33f4d..19d5536e 100644 --- a/scipy-stubs/optimize/_lsap.pyi +++ b/scipy-stubs/optimize/_lsap.pyi @@ -1,3 +1,7 @@ -from scipy._typing import Untyped +import numpy as np +import optype.numpy as onp -def linear_sum_assignment(*args: Untyped, **kwargs: Untyped) -> Untyped: ... +def linear_sum_assignment( + cost_matrix: onp.ToFloat2D, + maximize: onp.ToBool = False, +) -> tuple[onp.Array1D[np.intp], onp.Array1D[np.intp]]: ...