1SrSSKJr SSKJr SSKJr SSKJr SSKJ r SSKJ r SSKJ r SS KJ r "S S \ R5r"S S \ R5rSSjrSSjr"SS\5rg)z'Utilities for dealing with ML jobs API.)encoding) list_pager)apis) arg_utils) exceptions) properties) resources)yamlc\rSrSrSrSrg)NoFieldsSpecifiedErrorzEError indicating that no updates were requested in a Patch operation.N__name__ __module__ __qualname____firstlineno____doc____static_attributes__r,lib/googlecloudsdk/api_lib/ml_engine/jobs.pyr r sMrr c\rSrSrSrSrg)NoPackagesSpecifiedError z>Error that no packages were specified for non-custom training.rNrrrrrr sFrrc0[R"SU5$)Nml)rGetMessagesModule)versions rrr$s  g ..rc,[R"SXS9$)Nr)no_http)rGetClientInstance)rr s rr!r!(s  g ??rc\rSrSrSrSSjrSr\S5rSr \S5r \S 5r \S 5r SS jr S rS rSrSrSSjrSrSrSSjrSrg) JobsClient,z3Client for jobs service in the Cloud ML Engine API.NcU=(d [U5UlU=(d URRUlX0lgN)r!clientMESSAGES_MODULEmessages_short_message_prefix)selfr'r)short_message_prefixclient_versions r__init__JobsClient.__init__/s0=-n=DK; ; ;DM!5rc`[URSRURUS9S5$)Nz{prefix}{name})prefixname)getattrr)formatr*)r+short_message_names rGetShortMessageJobsClient.GetShortMessage5s9 4==#**$2L2L0B+DEI KKrcBURRR$r&)r)GoogleCloudMlV1JobStateValueValuesEnumr+s r state_enumJobsClient.state_enum:s == + + @ @@rcURRUR5S9n[R"UR R USSS9$)N)parentjobspageSize)fieldbatch_size_attribute)r)MlProjectsJobsListRequest RelativeNamer YieldFromListr' projects_jobs)r+ project_refreqs rListJobsClient.List>sN -- 1 1'') 2 +C  # # !!3f' ))rc.URR$r&)r)r9r;s r job_classJobsClient.job_classEs == + ++rc.URR$r&)r)GoogleCloudMlV1TrainingInputr;s rtraining_input_classJobsClient.training_input_classIs == 5 55rc.URR$r&)r)GoogleCloudMlV1PredictionInputr;s rprediction_input_class!JobsClient.prediction_input_classMs == 7 77rc6URRUUS9$)N)r?googleCloudMlV1Job)r)MlProjectsJobsCreateRequest)r+r?jobs r_MakeCreateRequestJobsClient._MakeCreateRequestQs$ == 4 4 5  rcURRRURUR 5US95$)N)r?rZ)r'rGCreater[rE)r+rHrZs rr^JobsClient.CreateVsA ;; $ $ + + ++-   rcURRUR5S9nURRR U5$)zCancels given job.r2)r)MlProjectsJobsCancelRequestrEr'rGCancelr+job_refrIs rrcJobsClient.Cancel\s= -- 3 39M9M9O 3 PC ;; $ $ + +C 00rcURRUR5S9nURRR U5$)Nra)r)MlProjectsJobsGetRequestrEr'rGGetrds rriJobsClient.Getas= -- 0 0g6J6J6L 0 MC ;; $ $ ( ( --rcjUR5n/nUR(a"URUlURS5 U(d [ S5eUR R UR5USRU5S9nURRRU5$)z Update a job.labelszNo updates requested.,)r2rX updateMask) rM needs_updaterlappendr r)MlProjectsJobsPatchRequestrEjoinr'rGPatch)r+re labels_updaterZ update_maskrIs rrsJobsClient.Patches .. CK!! ''cj"  "#: ;; -- 2 2  ! ! #88K( 3 C ;; $ $ * *3 //rc ,UR5nU(a>[R"U5nU(a![R"UUR5nU(aUUlU bU UlUR(dUR5UlUUUUUUU U U U US. nUR5H"unnUcM [URUU5 M$ U(a[R"USU5 U(Ga^UR5R5GH;unnUcM URS5(GaURS5(dUS(a*[R"USRU5US5 USS(a-[R"US RU5USS5 USS (a-[R"US RU5USS 5 US :Xa:US (a-[R"USRU5US 5 GMGM!GM$[URUU5 GM> UR!U5(d!UR#U5(d [%S5eU$)aBuilds a Cloud ML Engine Job from a config file and/or flag values. Args: path: path to a yaml configuration file module_name: value to set for moduleName field (overrides yaml file) job_name: value to set for jobName field (overrides yaml file) trainer_uri: List of values to set for trainerUri field (overrides yaml file) region: compute region in which to run the job (overrides yaml file) job_dir: Cloud Storage working directory for the job (overrides yaml file) scale_tier: ScaleTierValueValuesEnum the scale tier for the job (overrides yaml file) user_args: [str]. A list of arguments to pass through to the job. (overrides yaml file) runtime_version: the runtime version in which to run the job (overrides yaml file) python_version: the Python version in which to run the job (overrides yaml file) network: user network to which the job should be peered with (overrides yaml file) service_account: A service account (email address string) to use for the job. labels: Job.LabelsValue, the Cloud labels for the job kms_key: A customer-managed encryption key to use for the job. custom_train_server_config: jobs_util.CustomTrainingInputServerConfig, configuration object for custom server parameters. enable_web_access: whether to enable the interactive shell for the job. Raises: NoPackagesSpecifiedError: if a non-custom job was specified without any trainer_uris. Returns: A constructed Job object. ) pythonModuleargs packageUrisregionjobDir scaleTierruntimeVersion pythonVersionnetworkserviceAccountenableWebAccessz)trainingInput.encryptionConfig.kmsKeyNameConfigTfConfigimageUriztrainingInput.{}.imageUriacceleratorConfigtypez'trainingInput.{}.acceleratorConfig.typecountz(trainingInput.{}.acceleratorConfig.count workerConfig tpuTfVersionztrainingInput.{}.tpuTfVersionz#Non-custom jobs must have packages.)rMr load_pathr DictToMessagejobIdrl trainingInputrQitemssetattrrSetFieldInMessage GetFieldMapendswithr4HasPackageURIsIsCustomContainerTrainingr)r+path module_namejob_name trainer_urir{job_dir scale_tier user_argsruntime_versionpython_versionrservice_accountrlkms_keycustom_train_server_configenable_web_accessrZdataadditional_fields field_namevalues rBuildTrainingJobJobsClient.BuildTrainingJobusef .. C ^^D !d $$T4>>:ci cj   335c#")'), /446 E  !!:u57!!#"M")+"9EEGMMO *e  !!(++%%j11Z ))-44Z@ #%()&1));BB:N+,V46()'2))<CCJO+,W57^+n0E))188D')1F+ C%%z5 93 P6   s # #D,J,J3,O,O $%J KK Jrc@[URR5$r&)boolrrzr+rZs rrJobsClient.HasPackageURIss !!-- ..rc[URR=(a URRR5$r&)rr masterConfigrrs rr$JobsClient.IsCustomContainerTrainings7 !!..8!!..77 99rc F[RRRR 5nU(aUR S5nU"UUS9nOSnUR UUUU U U US9nURU5UlU(aUUl OuU(a8[RRUSUUS.S9nUR5UlO6[RRUSS U0S9nUR5UlU (aU UlUR#UUU S 9$) a"Builds a Cloud ML Engine Job for batch prediction from flag values. Args: job_name: value to set for jobName field model_dir: str, Google Cloud Storage location of the model files model_name: str, value to set for modelName field version_name: str, value to set for versionName field input_paths: list of input files data_format: format of the input files output_path: single value for the output location region: compute region in which to run the job runtime_version: the runtime version in which to run the job max_worker_count: int, the maximum number of workers to use batch_size: int, the number of records per batch sent to Tensorflow signature_name: str, name of input/output signature in the TF meta graph labels: Job.LabelsValue, the Cloud labels for the job accelerator_count: int, The number of accelerators to attach to the machines accelerator_type: AcceleratorsValueListEntryValuesEnum, The type of accelerator to add to machine. Returns: A constructed Job object. AcceleratorConfig)rrN) inputPaths outputPathr{r~maxWorkerCount batchSize acceleratorzml.projects.models.versions)modelsId projectsId) collectionparamszml.projects.modelsr)rpredictionInputrl)rVALUEScoreproject GetOrFailr6rUDataFormatValueValuesEnum dataFormaturir REGISTRYParserE versionName modelName signatureNamerM)r+r model_dir model_name version_name input_paths data_format output_pathr{rmax_worker_count batch_sizesignature_namerlaccelerator_countaccelerator_type project_idaccelerator_config_msgaccelerator_configprediction_input version_ref model_refs rBuildBatchPredictionJob"JobsClient.BuildBatchPredictionJobsMN""''//99;J#334GH18I7GI 22&'&3#3"L"L#& &&,, #@( C-Ek&1%=%=%?"$$** !5 ++-i$-#9#9#; '5$ >>(  r)r*r'r))NNGoogleCloudMlV1v1)NN)NNNNNNNNNNNNNNNN)NNNNNNNNNNNNNNN)rrrrrr.r6propertyr<rJrMrQrUr[r^rcrirsrrrrrrrrr#r#,s;+/FJ6 K  A A) , , 6 6 8 8  1 .0"!#' $#'"#"&!%'+&*#'+"#26)-!xt/9 (,(,)-+/*.*.*.%).2/3)--1%)04/3Prr#N)r)rF)rapitools.base.pyrrgooglecloudsdk.api_lib.utilr$googlecloudsdk.command_lib.util.apisrgooglecloudsdk.corerrr r Errorr rrr!objectr#rrrrse.&',:**)$NZ--NGz//G/@ZZr