
                            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 rS r\R(                  " \R*                  R,                  5       " S S\R.                  5      5       r\R(                  " \R*                  R2                  \R*                  R4                  5       " S S\R.                  5      5       rg)z#Vertex AI endpoints update command.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)errors)flags)labels_util)logc                     [         R                  " U S5        [         R                  " SSS9R                  U 5        [         R                  " S5      R                  U 5        [
        R                  " U 5        g )Nz	to updatetensorboardF)required)r   AddTensorboardResourceArgGetDisplayNameArgAddToParserGetDescriptionArgr   AddUpdateLabelsFlagsparsers    %lib/surface/ai/tensorboards/update.py_AddArgsr      sT    !!&+6-%8DDVL-(44V<""6*    c                 @  ^  T R                   R                  R                  5       nUR                  5       S   T l        [
        R                  " UT R                  S9   [        R                  " US9n UR                  UT 5      n[        R                  " UR                  SS9  UsSSS5        $ ! [        R                   aI    / SQn[        U 4S jU 5       5      (       d  e [        R                   R#                  S	5         SSS5        gf = f! , (       d  f       g= f)
)Update an existing Vertex AI Tensorboard.locationsId)region)versionzVertex AI Tensorboard)kindN)display_namedescriptionupdate_labelsclear_labelsremove_labelsc              3   F   >#    U  H  nTR                  U5      v   M     g 7fN)IsSpecified).0argargss     r   	<genexpr>_Run.<locals>.<genexpr>8   s!     H2G3!!#&&2Gs   !zNo update to perform.)CONCEPTSr   ParseAsDictr   r	   AiplatformEndpointOverridesr   TensorboardsClientPatchr   UpdatedResourcenamer
   NoFieldsSpecifiedErroranystatusPrint)r+   r   tensorboard_reftensorboards_clientopavailable_update_argss   `     r   _Runr>   &   s     MM--335/&&(7$+00M 33GD$$_d;b 
"''(?@' NM
 ((  H2GHHH	jj./! NM
 NMs1   D3B/ D/ADDDD
Dc                   6    \ rS rSrSrSS0r\S 5       rS rSr	g)	UpdateGaA   r   EXAMPLES            To update a Tensorboard `12345`, in region `us-central1` and project `my-project`, with the display name `updated display name`:

              $ {command} projects/my-project/locations/us-central1/tensorboards/12345 --display-name="updated display name"
          c                     [        U 5        g r'   r   r   s    r   ArgsUpdateGa.ArgsN   
    Vr   c                 6    [        U[        R                  5      $ r'   )r>   r   
GA_VERSIONselfr+   s     r   RunUpdateGa.RunR   s    i**++r    N
__name__
__module____qualname____firstlineno____doc__detailed_helpstaticmethodrF   rM   __static_attributes__rO   r   r   r@   r@   A   s/    1 -  ,r   r@   c                   6    \ rS rSrSrSS0r\S 5       rS rSr	g)	
UpdateBetaV   r   rB   rC   c                     [        U 5        g r'   rE   r   s    r   rF   UpdateBeta.Argsc   rH   r   c                 6    [        U[        R                  5      $ r'   )r>   r   BETA_VERSIONrK   s     r   rM   UpdateBeta.Rung   s    i,,--r   rO   NrP   rO   r   r   rZ   rZ   V   s/    1 -  .r   rZ   N)rU   
__future__r   r   r   &googlecloudsdk.api_lib.ai.tensorboardsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   r   $googlecloudsdk.command_lib.util.argsr   googlecloudsdk.corer   r   r>   ReleaseTracksReleaseTrackGAUpdateCommandr@   BETAALPHArZ   rO   r   r   <module>rm      s    * &  ' 9 ( 3 7 0 / < #+6 D%%((),t!! , *,( D%%**D,=,=,C,CD.## . E.r   