=SrSSKJr SSKJr SSKJr SSKJr SSKJr SSKJ r SSK J r SS K Jr \ R4S jr"S S \5rg )z4Utilities for querying hptuning-jobs in AI platform.)absolute_import)division)unicode_literals) list_pager)apis)messages) constants)yamlc[R"[R[RU5nU[R :XaUR R$URR$N) rGetMessagesModuler AI_PLATFORM_API_NAMEAI_PLATFORM_API_VERSION GA_VERSION GoogleCloudAiplatformV1StudySpecAlgorithmValueValuesEnum%GoogleCloudAiplatformV1beta1StudySpec)versionrs 6lib/googlecloudsdk/api_lib/ai/hp_tuning_jobs/client.pyGetAlgorithmEnumrs[  # #I$B$B$-$E$Eg$NP( $$$  4 4 M MM  9 9 R RRcr\rSrSrSrSrSrSrSrSSjr SS jr SS jr SS jr S r S rg)HpTuningJobsClient%zBClient used for interacting with HyperparameterTuningJob endpoint.c[R"[R[RU5nUR UlURUlXl [RUUl gr ) rGetClientInstancer rrMESSAGES_MODULE _messages+projects_locations_hyperparameterTuningJobs_servicerAI_PLATFORM_MESSAGE_PREFIX_message_prefix)selfrclients r__init__HpTuningJobsClient.__init__(s[  # #I$B$B$-$E$Eg$NPF++DNFFDML$??HDrc`[URSRURUS9S5$)z(Returns the API messsages class by name.z{prefix}{name})prefixnameN)getattrrformatr")r# message_names r _GetMessageHpTuningJobsClient._GetMessage0s9  t';';%1  348 ::rc$URS5$)z5Returns the HyperparameterTuningJob resource message.HyperparameterTuningJob)r-r#s rHyperparameterTuningJobMessage1HpTuningJobsClient.HyperparameterTuningJobMessage8s   5 66rc8URS5R$)z,Returns enum message representing Algorithm. StudySpec)r-rr1s r AlgorithmEnum HpTuningJobsClient.AlgorithmEnum=s   K ( A AArNc UR5n U(aA[R"U5nU(a$[R"XR55n U R (dU(dSU lO U(aXMlU R (dU(dSU lO U(aX]lU(aXRlU (aXRl U (aXRl U (aXRl U(aX-l U(a!U R(aXmRlUbURS5"US9U lU (aXlUR$[&R(:XaUR*R-UU S9nOUR*R-UU S9nUR.R1U5$)aCreates a hyperparameter tuning job with given parameters. Args: config_path: str, the file path of the hyperparameter tuning job configuration. display_name: str, the display name of the created hyperparameter tuning job. parent: str, parent of the created hyperparameter tuning job. e.g. /projects/xxx/locations/xxx/ max_trial_count: int, the desired total number of Trials. The default value is 1. parallel_trial_count: int, the desired number of Trials to run in parallel. The default value is 1. algorithm: AlgorithmValueValuesEnum, the search algorithm specified for the Study. kms_key_name: str, A customer-managed encryption key to use for the hyperparameter tuning job. network: str, user network to which the job should be peered with (overrides yaml file) service_account: str, A service account (email address string) to use for the job. enable_web_access: bool, Whether to enable the interactive shell for the job. enable_dashboard_access: bool, Whether to enable the dashboard defined for the job. labels: LabelsValues, map-like user-defined metadata to organize the hp-tuning jobs. Returns: Created hyperparameter tuning job. EncryptionSpec) kmsKeyName)parent.googleCloudAiplatformV1HyperparameterTuningJob)r<3googleCloudAiplatformV1beta1HyperparameterTuningJob)r2r load_path messages_utilDictToMessageWithErrorCheck maxTrialCountparallelTrialCount trialJobSpecnetworkserviceAccountenableWebAccessenableDashboardAccess displayName studySpec algorithmr-encryptionSpeclabelsrr rr@AiplatformProjectsLocationsHyperparameterTuningJobsCreateRequestr Create)r# config_path display_namer<max_trial_countparallel_trial_countrK kms_key_namerEservice_accountenable_web_accessenable_dashboard_accessrMjob_specdatarequests rrOHpTuningJobsClient.CreateBssZ224H ^^K (d  << 5579  ! !/ h .  & &/C$%h! $8!&-#-<*.?+4K1)X''%." $ 0 01A B!!#ho ||y+++__9A`Cg__>F`Hg ==   ((rcjURRUS9nURRU5$Nr)r=AiplatformProjectsLocationsHyperparameterTuningJobsGetRequestr Getr#r)rZs rraHpTuningJobsClient.Gets4nnZZ [G ==  W %%rcjURRUS9nURRU5$r])r@AiplatformProjectsLocationsHyperparameterTuningJobsCancelRequestr Cancelrbs rrfHpTuningJobsClient.Cancels4nn]] ^G ==   ((rct[R"URURR US9SSUS9$)N)r<hyperparameterTuningJobspageSize)fieldbatch_size_attributelimit)r YieldFromListr r>AiplatformProjectsLocationsHyperparameterTuningJobsListRequest)r#rmregions rListHpTuningJobsClient.Lists@  # #   G G H (' rc|^URRUS9nURRU5mU4SjnU$)zReturns a function to decide if log fetcher should continue polling. Args: name: String id of job. Returns: A one-argument function decides if log fetcher should continue. r^c.>US::agTRSL$)Nr9T)endTime)periods_without_logsresponses rShouldContinue;HpTuningJobsClient.CheckJobComplete..ShouldContinues  "    %%rr_)r#r)rZrxrws @rCheckJobComplete#HpTuningJobsClient.CheckJobCompletesCnnZZ [G}}  )H& r)r"rr r) NNNNNNNFFNr )NN)__name__ __module__ __qualname____firstlineno____doc__r%r-r2r6rOrarfrqrz__static_attributes__rrrr%sVJI:7 B# _)B& ) rrN)r __future__rrrapitools.base.pyrgooglecloudsdk.api_lib.utilrrr@googlecloudsdk.command_lib.air googlecloudsdk.corer BETA_VERSIONrobjectrrrrrsA;&'',A3$'33Sddr