
                         J   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dlmZ d Zd Z e
j*                  e
j,                  j.                  e
j,                  j0                         G d de
j2                               Zy)z1Vertex AI Tensorboard time series create command.    )absolute_import)division)unicode_literals)encoding)client)base)	constants)endpoint_util)flags)tensorboards_util)
validation)logc                    t        j                  | d       t        j                  dd      j                  |        t        j                  d      j                  |        t        j                  d      j                  j                  |        t        j                  d      j                  |        t        j                  d      j                  |        y )Nz#to create a Tensorboard time seriesztensorboard-time-seriesT)required)
r   AddTensorboardRunResourceArgGetDisplayNameArgAddToParserGetDescriptionArgr   GetTensorboardTimeSeriesTypeArg
choice_argGetPluginNameArgGetPluginDataArgparsers    0lib/surface/ai/tensorboard_time_series/create.py_AddArgsr       s    $$V%JL$00;F0C34@@H33!!+KK,?23??G23??G    c                    t        j                  | j                         | j                  j                  j                         }|j                         d   }t        j                  ||      5  t        j                  |      }|j                  ||       }t        j                  |      }d|v r1t        j                  j!                  dj#                  |d                |cddd       S # 1 sw Y   yxY w)/Create a new Vertex AI Tensorboard time series.locationsId)region)versionnamez.Created Vertex AI Tensorboard time series: {}.N)r   ValidateDisplayNamedisplay_nameCONCEPTStensorboard_runParseAsDictr
   AiplatformEndpointOverridesr   TensorboardTimeSeriesClientCreater   MessageToPyValuer   statusPrintformat)argsr"   tensorboard_run_refr!   tensorboard_runs_clientresponseresponse_msgs          r   _Runr6   ,   s      !2!2355;;=%%'6&00H$@@&--.A4HH,,X6L	jj;
C
C6"$%  IHHs   .A4C,,C5c                   .    e Zd ZdZddiZed        Zd Zy)
CreateBetar   EXAMPLESaL            To create a Tensorboard Time Series in Tensorboard `12345`, Tensorboard Experiment `my-tensorboard-experiment, and in Tensorboard Run `my-tensorboard-run`, with the display name `my tensorboard time series` and type Scalar:

              $ {command} my-tensorboard-run --tensorboard-experiment-id=my-tensorboard-experiment --tensorboard-id=12345 --display-name="my tensorboard time series" --type=SCALAR

          You may also provide a description:

              $ {command} my-tensorboard-run --tensorboard-experiment-id=my-tensorboard-experiment --tensorboard-id=12345 --display-name="my tensorboard time series" --type=SCALAR --description="my description"

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

              $ {command} projects/my-project/locations/us-central1/tensorboards/12345/experiments/my-tensorboard-experiment/runs/my-tensorboard-run --display-name="my tensorboard time series" --type=SCALAR
          c                     t        |        y N)r   r   s    r   ArgszCreateBeta.ArgsS   s
    Vr   c                 6    t        |t        j                        S r;   )r6   r	   BETA_VERSION)selfr1   s     r   RunzCreateBeta.RunW   s    i,,--r   N)__name__
__module____qualname____doc__detailed_helpstaticmethodr<   r@    r   r   r8   r8   >   s/    7 -"  .r   r8   N)rD   
__future__r   r   r   apitools.base.pyr   1googlecloudsdk.api_lib.ai.tensorboard_time_seriesr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air	   r
   r   r   r   googlecloudsdk.corer   r   r6   ReleaseTracksReleaseTrackBETAALPHACreateCommandr8   rG   r   r   <module>rS      s    8 &  ' % D ( 3 7 / ; 4 #	H$ D%%**D,=,=,C,CD.## . E.r   