
    N	                        S r SSKJr  SSKJr  SSKJr  SSKJr  SSKJr  SSKJ	r	  SSK
Jr  SS	K
Jr  SS
K
Jr  SSK
Jr  SSKJr  SSKJr  S rS r\	R*                  " \	R,                  R.                  \	R,                  R0                  5       " S S\	R2                  5      5       r\	R*                  " \	R,                  R6                  5       " S S\	R2                  5      5       rg)z"ai-platform models update command.    )absolute_import)division)unicode_literals)models)
operations)base)endpoint_util)flags)models_util)region_util)labels_util)logc                    [         R                  " 5       R                  U 5        [         R                  " SS9R                  U 5        [         R                  " S5      R                  U 5        [
        R                  " U 5        g)z:Get arguments for the `ai-platform models update` command.T)include_globalmodelN)r
   GetModelNameAddToParserGetRegionArgGetDescriptionFlagr   AddUpdateLabelsFlagsparsers    (lib/surface/ai_platform/models/update.py_AddUpdateArgsr       sY    ""6*D)55f=7#//7""6*    c                 L   [         R                  " U 5      n[        R                  " US9   [        R
                  " 5       n[        R                  " 5       n[        R                  " X#U 5        [        R                  " U R                  SS9  S S S 5        g ! , (       d  f       g = f)N)regionzai platform model)kind)r   	GetRegionr	   MlEndpointOverridesr   ModelsClientr   OperationsClientr   Updater   UpdatedResourcer   )argsr   models_clientoperations_clients       r   _Runr(   (   sn      &&((7'')M"335}>

)<=	 877s   A!B
B#c                   .    \ rS rSrSr\S 5       rS rSrg)
UpdateBeta1   %Update an existing AI Platform model.c                     [        U 5        g Nr   r   s    r   ArgsUpdateBeta.Args5   
    6r   c                     [        U5        g r.   r(   selfr%   s     r   RunUpdateBeta.Run9   	    Jr    N	__name__
__module____qualname____firstlineno____doc__staticmethodr0   r7   __static_attributes__r:   r   r   r*   r*   1       - r   r*   c                   .    \ rS rSrSr\S 5       rS rSrg)r#   =   r,   c                     [        U 5        g r.   r/   r   s    r   r0   Update.ArgsA   r2   r   c                     [        U5        g r.   r4   r5   s     r   r7   
Update.RunE   r9   r   r:   Nr;   r:   r   r   r#   r#   =   rC   r   r#   N)r@   
__future__r   r   r    googlecloudsdk.api_lib.ml_enginer   r   googlecloudsdk.callioper   $googlecloudsdk.command_lib.ml_enginer	   r
   r   r   $googlecloudsdk.command_lib.util.argsr   googlecloudsdk.corer   r   r(   ReleaseTracksReleaseTrackALPHABETAUpdateCommandr*   GAr#   r:   r   r   <module>rV      s    ) &  ' 3 7 ( > 6 < < < #+> D%%++T->->-C-CD##  E D%%(()T  *r   