o
    /wiA                     @  sB   d dl mZ d dlZd dlmZ d dlmZ ddddddZdS )    )annotationsN)acqf)sample_normalized_paramsi   )	n_samplesrngacqf_paramsacqf.AcquisitionFunctionParamsr   intr   np.random.RandomState | Nonereturntuple[np.ndarray, float]c                C  s>   t || j|d}t| |}t|}||d d f || fS )N)r   )r   search_spacer   eval_acqf_no_gradnpargmax)r   r   r   xsresbest_i r   T/home/ubuntu/sommelier/.venv/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   optuna._gp.search_spacer   r   r   r   r   r   <module>   s    