o
    iJ                  	   @   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testtest_vertex_tuned_models_updatemodels/2171259487439028224zMy SupervisedTuningJob8)descriptiondefault_checkpoint_idmodelconfig404)name
parametersexception_if_mldev;test_vertex_tuned_models_update_with_http_options_in_method)r   r   http_options
test_tablezmodels.update)fileglobals_for_file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(   Y/home/ubuntu/.local/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$   TestTableItem_UpdateModelParametersr   list__annotations__setup__file__globals
pytestmarkmarkasyncior*   r-   r(   r(   r(   r)   <module>   sJ   
	

