>dZddlmZddlmZddlmZddlmZddlmZddl m Z ddl m Z dd l m Z dd l m Z dd lmZdd lmZd ZdZej(ej*j,ej*j.Gddej0Zy)z1Command to update a Tensorboard run in Vertex AI.)absolute_import)division)unicode_literals)client)base) constants) endpoint_util)errors)flags) labels_util)logctj|dtjddj|tjdj|t j |y)Nz to updateztensorboard runF)required)r AddTensorboardRunResourceArgGetDisplayNameArg AddToParserGetDescriptionArgr AddUpdateLabelsFlagsparsers )lib/surface/ai/tensorboard_runs/update.py_AddArgsrsV$$V[9+e<HHP+,88@ ""6*c0jjj}|jd_t j |j5 tj|j|}tj|jd|cdddS#tj$rCgd}tfd|Dstj j#d YdddywxYw#1swYyxYw) -Update an existing Vertex AI Tensorboard run. locationsId)region)versionzVertex AI Tensorboard run)kindN) display_name update_labels clear_labels remove_labels descriptionc3@K|]}j|ywN) IsSpecified).0argargss r z_Run..3sH2G3!!#&2GszNo update to perform.)CONCEPTStensorboard_runParseAsDictrr AiplatformEndpointOverridesrTensorboardRunsClientPatchr UpdatedResourcenamer NoFieldsSpecifiedErroranystatusPrint)r*rtensorboard_run_refopavailable_update_argss` r_Runr<&s 55;;=#**,];$+00M  ' ' 8 > > t %b "''(CD NM  ( (H2GH H  jj./ NM NMs1D !&B3"D 3A D >D D  D  Dc.eZdZdZddiZedZdZy) UpdateBetarEXAMPLESa9 To update Tensorboard Time Run `my-tensorboard-run` in Tensorboard `12345`, Tensorboard Experiment `my-tensorboard-experiment`, with the display name `updated display name`: $ {command} projects/my-project/locations/us-central1/tensorboards/12345/experiments/my-tensorboard-experiment/runs/my-tensorboard-run --display-name="updated display name" Or with flags: $ {command} my-tensorboard-run --tensorboard-id=12345 --tensorboard-experiment-id=my-tensorboard-experiment --display-name="updated display name" ct|yr&)rrs rArgszUpdateBeta.ArgsMs  Vrc6t|tjSr&)r<r BETA_VERSION)selfr*s rRunzUpdateBeta.RunQs i,, --rN)__name__ __module__ __qualname____doc__ detailed_help staticmethodrArErrr>r><s/5  -.rr>N)rI __future__rrr*googlecloudsdk.api_lib.ai.tensorboard_runsrgooglecloudsdk.calliopergooglecloudsdk.command_lib.airr r r $googlecloudsdk.command_lib.util.argsr googlecloudsdk.corer rr< ReleaseTracks ReleaseTrackBETAALPHA UpdateCommandr>rLrrrXs~8&'=(370/<#+,D%%**D,=,=,C,CD.##.E.r