y1 SrSSKJr SSKJr SSKJr SSKJr SSKJr SSKJ r SSK J r SS K J r SS K Jr SS K Jr SS K Jr "S S\ R"5r"SS\ R"5rSSjrSSjr"SS\5rg)z'Utilities for dealing with ML jobs API.)absolute_import)division)unicode_literals)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.pyrrsMrrc\rSrSrSrSrg)NoPackagesSpecifiedError#z>Error that no packages were specified for non-custom training.rNrrrrrr#sFrrc0[R"SU5$)Nml)rGetMessagesModule)versions rr r 's  g ..rc,[R"SXS9$)Nr)no_http)rGetClientInstance)r!r#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__2s0=-n=DK; ; ;DM!5rc`[URSRURUS9S5$)Nz{prefix}{name})prefixname)getattrr,formatr-)r.short_message_names rGetShortMessageJobsClient.GetShortMessage8s9 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.ListAsN -- 1 1'') 2 +C  # # !!3f' ))rc.URR$r))r,r<r>s r job_classJobsClient.job_classHs == + ++rc.URR$r))r,GoogleCloudMlV1TrainingInputr>s rtraining_input_classJobsClient.training_input_classLs == 5 55rc.URR$r))r,GoogleCloudMlV1PredictionInputr>s rprediction_input_class!JobsClient.prediction_input_classPs == 7 77rc6URRUUS9$)N)rBgoogleCloudMlV1Job)r,MlProjectsJobsCreateRequest)r.rBjobs r_MakeCreateRequestJobsClient._MakeCreateRequestTs$ == 4 4 5  rcURRRURUR 5US95$)N)rBr])r*rJCreater^rH)r.rKr]s rraJobsClient.CreateYsA ;; $ $ + + ++-   rcURRUR5S9nURRR U5$)zCancels given job.r5)r,MlProjectsJobsCancelRequestrHr*rJCancelr.job_refrLs rrfJobsClient.Cancel_s= -- 3 39M9M9O 3 PC ;; $ $ + +C 00rcURRUR5S9nURRR U5$)Nrd)r,MlProjectsJobsGetRequestrHr*rJGetrgs rrlJobsClient.Getds= -- 0 0g6J6J6L 0 MC ;; $ $ ( ( --rcjUR5n/nUR(a"URUlURS5 U(d [ S5eUR R UR5USRU5S9nURRRU5$)z Update a job.labelszNo updates requested.,)r5r[ updateMask) rP needs_updateroappendrr,MlProjectsJobsPatchRequestrHjoinr*rJPatch)r.rh labels_updater] update_maskrLs rrvJobsClient.Patchhs .. 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.)rPr load_pathr DictToMessagejobIdro trainingInputrTitemssetattrr SetFieldInMessage GetFieldMapendswithr7HasPackageURIsIsCustomContainerTrainingr)r.path module_namejob_name trainer_urir~job_dir scale_tier user_argsruntime_versionpython_versionrservice_accountrokms_keycustom_train_server_configenable_web_accessr]dataadditional_fields field_namevalues rBuildTrainingJobJobsClient.BuildTrainingJobxsef .. 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))boolrr}r.r]s 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~rmaxWorkerCount batchSize acceleratorzml.projects.models.versions)modelsId projectsId) collectionparamszml.projects.modelsr)rpredictionInputro)r VALUEScoreproject GetOrFailr9rXDataFormatValueValuesEnum dataFormaturir REGISTRYParserH versionName modelName signatureNamerP)r.r model_dir model_name version_name input_paths data_format output_pathr~rmax_worker_count batch_sizesignature_nameroaccelerator_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)rrrrrr1r9propertyr?rMrPrTrXr^rarfrlrvrrrrrrrrr&r&/s;+/FJ6 K  A A) , , 6 6 8 8  1 .0"!#' $#'"#"&!%'+&*#'+"#26)-!xt/9 (,(,)-+/*.*.*.%).2/3)--1%)04/3Prr&N)r)rF)r __future__rrrapitools.base.pyrrgooglecloudsdk.api_lib.utilr$googlecloudsdk.command_lib.util.apisr googlecloudsdk.corer r r r Errorrrr r$objectr&rrrrsl.&'%',:**)$NZ--NGz//G/@ZZr