From 9e0157f525cb4f69f2f3095f642c686bc72aa373 Mon Sep 17 00:00:00 2001
From: jorenham <jhammudoglu@gmail.com>
Date: Thu, 28 Nov 2024 04:57:56 +0100
Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=E2=9C=A8=20fix=20and=20complete=20?=
 =?UTF-8?q?`scipy.optimize.direct`?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 scipy-stubs/optimize/_direct_py.pyi | 55 ++++++++++++++---------------
 1 file changed, 27 insertions(+), 28 deletions(-)

diff --git a/scipy-stubs/optimize/_direct_py.pyi b/scipy-stubs/optimize/_direct_py.pyi
index 39a41746..a556c171 100644
--- a/scipy-stubs/optimize/_direct_py.pyi
+++ b/scipy-stubs/optimize/_direct_py.pyi
@@ -1,40 +1,39 @@
-from collections.abc import Callable, Iterable
-from typing import Any, type_check_only
+from collections.abc import Callable
+from typing import Concatenate
 
 import numpy as np
-import numpy.typing as npt
 import optype.numpy as onp
-from scipy._typing import Untyped
 from ._constraints import Bounds
-from ._optimize import OptimizeResult
+from ._optimize import OptimizeResult as _OptimizeResult
 
 __all__ = ["direct"]
 
-ERROR_MESSAGES: tuple[str, ...] = ...
-SUCCESS_MESSAGES: tuple[str, ...] = ...
-
-@type_check_only
-class _OptimizeResult(OptimizeResult):
-    message: str
-    success: bool
+class OptimizeResult(_OptimizeResult):
+    x: onp.Array1D[np.float64]
+    fun: float | np.float64
     status: int
-    fun: float
-    x: onp.ArrayND[np.float64]  # 1d
-    nit: int
+    success: bool
+    message: str
     nfev: int
+    nit: int
+
+###
+
+ERROR_MESSAGES: tuple[str, str, str, str, str, str, str, str, str, str, str] = ...
+SUCCESS_MESSAGES: tuple[str, str, str] = ...
 
 def direct(
-    func: Callable[[npt.ArrayLike, tuple[Any]], float],
-    bounds: Iterable[float] | Bounds,
+    func: Callable[Concatenate[onp.Array1D[np.float64], ...], onp.ToFloat],
+    bounds: tuple[onp.ToFloat1D, onp.ToFloat1D] | Bounds,
     *,
-    args: tuple[Untyped, ...] = (),
-    eps: float = 0.0001,
-    maxfun: int | None = None,
-    maxiter: int = 1000,
-    locally_biased: bool = True,
-    f_min: float = ...,
-    f_min_rtol: float = 0.0001,
-    vol_tol: float = 1e-16,
-    len_tol: float = 1e-06,
-    callback: Callable[[npt.ArrayLike], None] | None = None,
-) -> _OptimizeResult: ...
+    args: tuple[object, ...] = (),
+    eps: onp.ToFloat = 1e-4,
+    maxfun: onp.ToJustInt | None = None,
+    maxiter: onp.ToJustInt = 1000,
+    locally_biased: onp.ToBool = True,
+    f_min: onp.ToFloat = ...,
+    f_min_rtol: onp.ToFloat = 0.0001,
+    vol_tol: onp.ToFloat = 1e-16,
+    len_tol: onp.ToFloat = 1e-06,
+    callback: Callable[[onp.ToArray1D], None] | None = None,
+) -> OptimizeResult: ...