
                             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\5      rg)z6Utilities for AI Platform Tensorboard experiments API.    )absolute_import)division)unicode_literals)
list_pager)apis)common_args)	constants)errors)labels_utilc                   j    \ rS rSrSrSS\R                  4S jrS rS r	SS jr
S rS	 rS
 rS rSrg)TensorboardExperimentsClient   zEHigh-level client for the AI Platform Tensorboard experiment surface.Nc                    U=(       d6    [         R                  " [        R                  [        R                  U   5      U l        U=(       d    U R
                  R                  U l        U R
                  R                  U l	        X0l
        g N)r   GetClientInstancer	   AI_PLATFORM_API_NAMEAI_PLATFORM_API_VERSIONclientMESSAGES_MODULEmessages+projects_locations_tensorboards_experiments_service_version)selfr   r   versions       ?lib/googlecloudsdk/api_lib/ai/tensorboard_experiments/client.py__init__%TensorboardExperimentsClient.__init__    sb      4D22&&))'24DK ; ; ;DMKKKKDMM    c                 $    U R                  X5      $ r   )
CreateBeta)r   tensorboard_refargss      r   Create#TensorboardExperimentsClient.Create+   s    ???11r   c           	      `   [         R                  " X R                  R                  R                  5      nU R                  R                  UR                  5       U R                  R                  UR                  UR                  US9UR                  S9nU R                  R                  U5      $ )z$Create a new Tensorboard experiment.)displayNamedescriptionlabels)parent1googleCloudAiplatformV1beta1TensorboardExperimenttensorboardExperimentId)r   ParseCreateArgsr   1GoogleCloudAiplatformV1beta1TensorboardExperimentLabelsValue?AiplatformProjectsLocationsTensorboardsExperimentsCreateRequestRelativeNamedisplay_namer(   tensorboard_experiment_idr   r$   )r   r"   r#   r)   requests        r   r!   'TensorboardExperimentsClient.CreateBeta.   s    ((mmMM	F mm[[++-:>--	:	:))(( 
; 
 !% > > \ @G ==((r   c           	          U R                   R                  UR                  5       [        R                  " U5      S9n[
        R                  " U R                  USSUUS9$ )N)r*   orderBytensorboardExperimentspageSize)fieldbatch_size_attribute
batch_sizelimit)r   =AiplatformProjectsLocationsTensorboardsExperimentsListRequestr1   r   ParseSortByArgr   YieldFromListr   )r   r"   r=   	page_sizesort_byr4   s         r   List!TensorboardExperimentsClient.List=   sa    mmYY++-**73 Z 5G ##&' r   c                     U R                   R                  UR                  5       S9nU R                  R	                  U5      $ N)name)r   <AiplatformProjectsLocationsTensorboardsExperimentsGetRequestr1   r   Getr   tensorboard_exp_refr4   s      r   rI    TensorboardExperimentsClient.GetI   s=    mmXX --/ Y 1G==W%%r   c                     U R                   R                  UR                  5       S9nU R                  R	                  U5      $ rF   )r   ?AiplatformProjectsLocationsTensorboardsExperimentsDeleteRequestr1   r   DeleterJ   s      r   rO   #TensorboardExperimentsClient.DeleteN   sB    	H	H$113 
I 
5  ==((r   c                 $    U R                  X5      $ r   )	PatchBeta)r   rK   r#   s      r   Patch"TensorboardExperimentsClient.PatchU   s    >>-44r   c                   ^ ^ T R                   R                  5       n/ nU U4S jn[        R                  " UT R                   R                  R                  U5      nUR
                  (       a"  UR                  Ul        UR                  S5        UR                  b"  UR                  Ul	        UR                  S5        UR                  b"  UR                  Ul
        UR                  S5        U(       d  [        R                  " S5      eT R                   R                  TR                  5       USR                  U5      S9nT R                   R#                  U5      $ )z Update a Tensorboard experiment.c                  :   > T R                  T5      R                  $ r   )rI   r)   )r   rK   s   r   	GetLabels9TensorboardExperimentsClient.PatchBeta.<locals>.GetLabels^   s    XX)*111r   r)   r2   r(   zNo updates requested.,)rG   r+   
updateMask)r   r.   r   ProcessUpdateArgsLazyr/   needs_updater)   appendr2   r'   r(   r
   NoFieldsSpecifiedError>AiplatformProjectsLocationsTensorboardsExperimentsPatchRequestr1   joinr   rS   )r   rK   r#   tensorboard_expupdate_maskrW   labels_updater4   s   ``      r   rR   &TensorboardExperimentsClient.PatchBetaX   s$    	GGI K2  55dmmMM	i!M !!,33o"$$($5$5o!(#$($4$4o!'))*ABBmmZZ --/:I88K( [ *G ==w''r   )r   r   r   r   )i  2   N)__name__
__module____qualname____firstlineno____doc__r	   BETA_VERSIONr   r$   r!   rC   rI   rO   rS   rR   __static_attributes__ r   r   r   r      s>    M  --	2)
&
)5(r   r   N)rj   
__future__r   r   r   apitools.base.pyr   googlecloudsdk.api_lib.utilr   r   googlecloudsdk.command_lib.air	   r
   $googlecloudsdk.command_lib.util.argsr   objectr   rm   r   r   <module>rt      s3    = &  ' ' , 3 3 0 <Z(6 Z(r   