o
    }oi                     @  s6   d dl mZ d dlZd dlmZ ddddddZdS )    )annotationsN)acqfi   )	n_samplesrngr   acqf_module.BaseAcquisitionFuncr   intr   np.random.RandomState | Nonereturntuple[np.ndarray, float]c                C  s<   | j j||d}| |}t|}||d d f || fS )N)r   )search_spacesample_normalized_paramseval_acqf_no_gradnpargmax)r   r   r   xsresbest_i r   K/home/ubuntu/.local/lib/python3.10/site-packages/optuna/_gp/optim_sample.pyoptimize_acqf_sample   s   

r   )r   r   r   r   r   r   r	   r
   )
__future__r   numpyr   
optuna._gpr   acqf_moduler   r   r   r   r   <module>   s    