o
    YiJ                  	   @   s   U d dl Z ddlmZ ddlmZ ddlmZ dddiiZejd	ejd
ddddddejdejd
ddeddddgZe	ej e
d< ejee dedZe jjdd Ze jjdd ZdS )    N   )errors)types   )pytest_helperheaderstestZtest_vertex_tuned_models_updatemodels/2171259487439028224zMy SupervisedTuningJob8)descriptiondefault_checkpoint_idmodelconfig404)name
parametersZexception_if_mldevZ;test_vertex_tuned_models_update_with_http_options_in_method)r   r   http_options
test_tablezmodels.update)fileZglobals_for_fileZtest_methodr   c                    sl   | j jr2ttj}| jjjddt	ddI d H  W d    n1 s%w   Y  dt
|v s4J d S d S )Nz4tunedModels/generatenum5443-ekrw7ie9wis23zbeogbw6jq8My tuned gemini model)r   r   r   r   )_api_clientvertexaipytestraisesr   ClientErroraiomodelsupdatetest_http_optionsstr)cliente r#   _/home/kim/smarthome/.venv/lib/python3.10/site-packages/google/genai/tests/models/test_update.pytest_async_update_tuned_model>   s   	r%   c                    s   | j jr| jjjddtddI d H }d S ttj	}| jjjddtddI d H  W d    n1 s6w   Y  dt
|v sCJ d S )Nr	   r   )display_namer   r   r   )r   r   r   r   r   r   r   r   r   r   r    )r!   responser"   r#   r#   r$   test_async_update_modelL   s"   r(   )r    r   r   r   r   ZTestTableItemZ_UpdateModelParametersr   list__annotations__setup__file__globalsZ
pytestmarkmarkZasyncior%   r(   r#   r#   r#   r$   <module>   sJ   
	

