
    +                     >   d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddl	m
Z
 ddl	mZ dd	l	mZ dd
l	mZ ddlmZ ddlmZ d Zd Z ej(                  ej*                  j,                  ej*                  j.                         G d dej0                               Zy)z0Vertex AI Tensorboard experiment create command.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)
validation)labels_util)logc                 4   t        j                  | d       t        j                  dd      j                  |        t        j                  d      j                  |        t        j                  |        t        j                         j                  |        y )Nz"to create a Tensorboard experimentztensorboard-experimentF)required)r
   AddTensorboardResourceArgGetDisplayNameArgAddToParserGetDescriptionArgr   AddCreateLabelsFlagsGetTensorboardExperimentIdArgparsers    0lib/surface/ai/tensorboard_experiments/create.py_AddArgsr      so    !!&*NO00;F0C23??G""6*%%'33F;    c                    t        j                  | j                         | j                  j                  j                         }|j                         d   }t        j                  ||      5  t        j                  |      }|j                  ||       }|j                  r8t        j                  j                  dj!                  |j                               |cddd       S # 1 sw Y   yxY w).Create a new Vertex AI Tensorboard experiment.locationsId)region)versionz-Created Vertex AI Tensorboard experiment: {}.N)r   ValidateDisplayNamedisplay_nameCONCEPTStensorboardParseAsDictr	   AiplatformEndpointOverridesr   TensorboardExperimentsClientCreatenamer   statusPrintformat)argsr   tensorboard_refr   tensorboard_experiments_clientresponses         r   _Runr1   )   s      !2!23MM--335/!!#M2&00H%+%H%H&"-44_dKH}}	jjGOO
--  IHHs   .A.C&&C/c                   .    e Zd ZdZddiZed        Zd Zy)
CreateBetar   EXAMPLESa            To create a Tensorboard Experiment in a Tensorboard `12345`, with the display name `my tensorboard experiment`:

              $ {command} 12345 --tensorboard-experiment-id=my-tensorboard-experiment --display-name="my tensorboard experiment"

          You may also provide a description and/or labels:

              $ {command} 12345 --tensorboard-experiment-id=my-tensorboard-experiment --description="my description" --labels="label1=value1" --labels="label2=value2"

          To create a Tensorboard Experiment `my-tensorboard-experiment` in a Tensorboard `12345`, region `us-central1`, and project `my-project`:

              $ {command} projects/my-project/locations/us-central1/tensorboards/12345 --tensorboard-experiment-id=my-tensorboard-experiment
          c                     t        |        y N)r   r   s    r   ArgszCreateBeta.ArgsN   s
    Vr   c                 6    t        |t        j                        S r6   )r1   r   BETA_VERSION)selfr-   s     r   RunzCreateBeta.RunR   s    i,,--r   N)__name__
__module____qualname____doc__detailed_helpstaticmethodr7   r;    r   r   r3   r3   9   s/    6 -"  .r   r3   N)r?   
__future__r   r   r   1googlecloudsdk.api_lib.ai.tensorboard_experimentsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   r   $googlecloudsdk.command_lib.util.argsr   googlecloudsdk.corer   r   r1   ReleaseTracksReleaseTrackBETAALPHACreateCommandr3   rB   r   r   <module>rN      s~    7 &  ' D ( 3 7 / 4 < #<  D%%**D,=,=,C,CD.## . E.r   