
    
                     \   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K	Jr  SS	KJr  \R$                  " \R&                  R(                  \R&                  R*                  \R&                  R,                  5       " S S\R.                  5      5       rg)zICommand to check stream logs of a hyperparameter tuning job in Vertex AI.    )absolute_import)division)unicode_literals)client)base)	constants)endpoint_util)flags)log_util)
validationc                   .    \ rS rSrSr\S 5       rS rSrg)
StreamLogs   zStream logs from a running Vertex AI hyperparameter tuning job.

## EXAMPLES

To stream logs of a hyperparameter tuning job, run:

$ {command} stream-logs HP_TUNING_JOB
c                     [         R                  " U S5        [        R                  " U 5        U R                  R                  [        R                  5        g )Nzto fetch stream log)hp_tuning_job_flagsAddHptuningJobResourceArgcommon_flagsAddStreamLogsFlagsdisplay_info	AddFormatr   
LOG_FORMAT)parsers    ,lib/surface/ai/hp_tuning_jobs/stream_logs.pyArgsStreamLogs.Args+   s;    11&:OP##F+
!!("5"56    c           	         UR                   R                  R                  5       nUR                  5       S   n[        R
                  " U[        R                  S9  U R                  5       [        R                  R                  :X  a  [        R                  O[        R                  n[        R                  " XCS9   UR                  5       n[         R"                  " UR                  5       S   [$        R&                  " US9R)                  U5      UR*                  UR,                  UR.                  S9sS S S 5        $ ! , (       d  f       g = f)NlocationsId)available_regions)versionregionhyperparameterTuningJobsId)r    )continue_functionpolling_interval	task_nameallow_multiline)CONCEPTShptuning_jobParseAsDictr   ValidateRegionr   SUPPORTED_TRAINING_REGIONSReleaseTrackr   GA
GA_VERSIONBETA_VERSIONr	   AiplatformEndpointOverridesRelativeNamer   r   r   HpTuningJobsClientCheckJobCompleter$   r%   allow_multiline_logs)selfargshptuning_job_refr!   r    relative_names         r   RunStreamLogs.Run1   s   }}11779$$&}5F)"F"FH '+&7&7 '					'i""#,#9#9 		2	2
(&335m  

!
!
#$@
A"55//>00NN335
( 
( 
(s   ;A8D==
E N)	__name__
__module____qualname____firstlineno____doc__staticmethodr   r:   __static_attributes__r<   r   r   r   r      s      7 7
5r   r   N)rA   
__future__r   r   r   (googlecloudsdk.api_lib.ai.hp_tuning_jobsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air   r	   r
   r   r   r   ,googlecloudsdk.command_lib.ai.hp_tuning_jobsr   ReleaseTracksr-   r.   BETAALPHACommandr   r<   r   r   <module>rM      s    P &  ' ; ( 3 7 ? 2 4 U D%%(($*;*;*@*@%%++-!5 !5-!5r   