
                            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  SSKJr  SSKJr  SSKJr  Sr\R4                  " \R6                  R8                  5       " S S\R:                  5      5       r\R4                  " \R6                  R>                  \R6                  R@                  5       " S S\5      5       r!g)z;Command to create a hyperparameter tuning job in Vertex AI.    )absolute_import)division)unicode_literals)util)client)base)	constants)endpoint_util)
validation)flags)hp_tuning_jobs_util)	arg_utils)labels_util)logzHyperparameter tuning job [{id}] submitted successfully.

Your job is still active. You may view the status of your job with the command

  $ gcloud{command_version} ai hp-tuning-jobs describe {id} --region={region}

Job State: {state}c                   N    \ rS rSrSr\R                  rSS0r\	S 5       r
S rSrg)	CreateGa,   #Create a hyperparameter tuning job.EXAMPLESz          To create a job named ``test'' under project ``example'' in region
          ``us-central1'', run:

            $ {command} --region=us-central1 --project=example --config=config.yaml --display-name=test
          c                     [         R                  " U[        R                  " SU R                  S9R
                  5        g )N	StudySpecversion)r   AddCreateHpTuningJobFlagsapi_util
GetMessage_api_versionAlgorithmValueValuesEnum)clsparsers     'lib/surface/ai/hp_tuning_jobs/create.pyArgsCreateGa.Args<   s3    	##K$'$4$4	66N6NP    c                 8   UR                   R                  R                  5       nUR                  5       S   n[        R
                  " U[        R                  S9  [        R                  " U R                  US9   [        R                  " U R                  S9n[        R                  " UR                  UR!                  5       5      n["        R$                  " UUR'                  5       R(                  5      nUR+                  UR-                  5       UR.                  UR0                  UR2                  UR4                  U[        R6                  " U5      UR8                  UR:                  UR<                  UR>                  US9n[@        RB                  RE                  [F        RI                  [J        RL                  " URN                  5      [J        RP                  " U RS                  5       5      UURT                  S95        UsS S S 5        $ ! , (       d  f       g = f)NlocationsId)available_regions)r   regionr   )parentconfig_pathdisplay_namemax_trial_countparallel_trial_count	algorithmkms_key_namenetworkservice_accountenable_web_accessenable_dashboard_accesslabels)idcommand_versionr(   state)+CONCEPTSr(   ParseAsDictr   ValidateRegionr	   SUPPORTED_TRAINING_REGIONSr
   AiplatformEndpointOverridesr   r   HpTuningJobsClientr   ChoiceToEnumr.   AlgorithmEnumr   ParseCreateArgsHyperparameterTuningJobMessageLabelsValueCreateRelativeNameconfigr+   r,   r-   GetAndValidateKmsKeyr0   r1   r2   r3   r   statusPrint&_HPTUNING_JOB_CREATION_DISPLAY_MESSAGEformatr   ParseJobNamenameOutputCommandVersionReleaseTrackr7   )selfargs
region_refr(   
api_clientr.   r4   responses           r!   RunCreateGa.RunC   s   %%++-J /F)"F"FH 
	2	2!!&
2,,T5F5FGj(()3)A)A)CEi**


3
3
5
A
ACf ""((*kk((..#88!66t<,,.. 22"&">"> # h 
jj
0
7
7$11(--@1FF##%'NN 8 $% =
2 
2 
2s   :FH
H N)__name__
__module____qualname____firstlineno____doc__r	   
GA_VERSIONr   detailed_helpclassmethodr"   rU   __static_attributes__rW   r$   r!   r   r   ,   s<    +%%, - P P$r$   r   c                   0    \ rS rSrSr\R                  rSrg)CreatePreGaj   r   rW   N)	rX   rY   rZ   r[   r\   r	   BETA_VERSIONr   r`   rW   r$   r!   rb   rb   j   s    +'',r$   rb   N)"r\   
__future__r   r   r   googlecloudsdk.api_lib.air   r   (googlecloudsdk.api_lib.ai.hp_tuning_jobsr   googlecloudsdk.callioper   googlecloudsdk.command_lib.air	   r
   r   ,googlecloudsdk.command_lib.ai.hp_tuning_jobsr   r   $googlecloudsdk.command_lib.util.apisr   $googlecloudsdk.command_lib.util.argsr   googlecloudsdk.corer   rJ   ReleaseTracksrO   GACreateCommandr   BETAALPHArb   rW   r$   r!   <module>rs      s    B &  ' 6 ; ( 3 7 4 > L : < #* & D%%(():t!! : *:z D%%**D,=,=,C,CD(( ( E(r$   