
    
                     D    S r SSKJr  SSKJr  SSKJr  S rSSSS	.rS
 r	g)z/Utilities for AI Platform Tensorboard commands.    )client)	arg_utils)	resourcesc                     SnSU ;   a  US-  nSU ;   a  US-  nSU ;   a  US-  nUS-  n [         R                  R                  XS	9$ ! [         R                   a!    [         R                  R                  U S
S	9s $ f = f)zParse operation relative resource name to the operation reference object.

Args:
  operation_name: The operation resource name

Returns:
  The operation reference object
zaiplatform.projects.locationsz/tensorboards/z.tensorboardsz/experiments/z.experimentsz/runs/z.runsz.operations)
collectionz(aiplatform.projects.locations.operations)r   REGISTRYParseRelativeName WrongResourceCollectionException)operation_namer   s     6lib/googlecloudsdk/command_lib/ai/tensorboards_util.pyParseTensorboardOperationr      s     /*'/!J&. J'J*O// 0 / /		3	3 O//#M 0 O OOs   A 2A<;A<)scalarzfUsed for tensorboard-time-series that is a list of scalars. E.g. accuracy of a model over epochs/time.)tensorz{Used for tensorboard-time-series that is a list of tensors. E.g. histograms of weights of layer in a model over epoch/time.)zblob-sequencezzUsed for tensorboard-time-series that is a list of blob sequences. E.g. set of sample images with labels over epochs/time.)SCALARTENSORBLOB_SEQUENCEc           
          [         R                  " S[        R                  " 5       R                  R
                  S[        SR                  U S9S S9$ )Nz--typeTzValue type of the {noun}.noun)requiredcustom_mappingshelp_strdefault)r   ChoiceEnumMapperr   GetMessagesModule1GoogleCloudAiplatformV1beta1TensorboardTimeSeriesValueTypeValueValuesEnum_TYPE_CHOICESformatr   s    r   GetTensorboardTimeSeriesTypeArgr    C   sL    		#	# 99#*11t1<
     N)
__doc__1googlecloudsdk.api_lib.ai.tensorboard_time_seriesr   $googlecloudsdk.command_lib.util.apisr   googlecloudsdk.corer   r   r   r     r!   r   <module>r'      s7    6 E : )O4

&	r!   