ASrSSKJr SSKJr SSKJr SSKJr SSKJr SSKJ r SSK J r SS K J r SS K Jr SS Kr"S S \ R 5r"SS\ R 5rSSjrSSjr"SS\5rg )z+Utilities for dealing with ML versions API.)absolute_import)division)unicode_literals)encoding) list_pager)apis) exceptions)yaml)textNc\rSrSrSrSrg)InvalidVersionConfigFile7Error indicating an invalid Version configuration file.N__name__ __module__ __qualname____firstlineno____doc____static_attributes__r4lib/googlecloudsdk/api_lib/ml_engine/versions_api.pyr r ?rr c\rSrSrSrSrg)NoFieldsSpecifiedError"rrNrrrrrr"rrrc0[R"SU5$)Nml)rGetMessagesModule)versions rr r &s  g ..rc,[R"SXS9$)Nr)no_http)rGetClientInstance)r!r#s rr$r$*s  g ??rc\rSrSrSr\"/SQ5r\"SS/5rSSjr\ S5r S r S r S r SS jrS rSrSrSrSrSrSrSSjrSrg)VersionsClient.z3Client for the versions service of Cloud ML Engine.) autoScaling deploymentUri description frameworklabels machineType manualScaling packageUrispredictionClass pythonVersionruntimeVersionserviceAccount containerroutesNc|U=(d [5UlU=(d URRUlgN)r$clientMESSAGES_MODULEmessages)selfr8r:s r__init__VersionsClient.__init__Es(/-/DK; ; ;DMrc.URR$r7)r:GoogleCloudMlV1Version)r;s r version_classVersionsClient.version_classIs == / //rc6URRUUS9$)N)parentgoogleCloudMlV1Version)r:%MlProjectsModelsVersionsCreateRequest)r;rCr!s r_MakeCreateRequest!VersionsClient._MakeCreateRequestMs$ == > >& ? ((rcjURR5nURRUUS9$)N)name'googleCloudMlV1SetDefaultVersionRequest)r:'GoogleCloudMlV1SetDefaultVersionRequest)MlProjectsModelsVersionsSetDefaultRequest)r;rIrequests r_MakeSetDefaultRequest%VersionsClient._MakeSetDefaultRequestRs7mmCCEG == B B 07 C 99rcURRRURUR 5US95$)z+Creates a new version in an existing model.)rCr!)r8projects_models_versionsCreaterF RelativeName)r; model_refr!s rrRVersionsClient.CreateXsA ;; / / 6 6 ))+   rc URR5n /n UR(a"URU lU R S5 U(aX;lU R S5 Ub3UR(a"U R S5 UR U lUbU R S5 X[lUb/U R S5 URRUS9U l Ub/U R S5 URRUS9U l UbMU R S 5 U RbXRl OURRUS 9U l U b/U R S 5 URRU S 9U lU bFS U ;a0U R S 5 URRU S 9U lOXRlU (d [#S5eUR$R&R)URR+UR-5U SR/[1U 55S95$)zUpdate a version.r,r*r0r/zmanualScaling.nodes)nodeszautoScaling.minNodes)minNodeszautoScaling.maxNodes)maxNodesrequestLoggingConfig)bigqueryTableName)samplingPercentagezNo updates requested.,)rIrD updateMask)r:r? needs_updater,appendr*valuer0r/GoogleCloudMlV1ManualScalingr.GoogleCloudMlV1AutoScalingr(rY#GoogleCloudMlV1RequestLoggingConfigrZr\rr8rQPatch$MlProjectsModelsVersionsPatchRequestrSjoinsorted) r; version_ref labels_updater*prediction_class_update package_urismanual_scaling_nodesauto_scaling_min_nodesauto_scaling_max_nodessampling_percentagebigquery_table_namer! update_masks rreVersionsClient.Patch_s? mm224GK!!$++gn"''*/F/S/S*+ 7 = =g'('./"mmHH$I&g)/0 MMDD)E+g)/0    ('=$"mmFF+G-&/0%)]]%V%V/&W&1g"& { 212'+}}'X'X2(Y(4$;N$$7  "#: ;; ;; / / 5 5 ::))+#*xx{ 34 ; 6 77rcURRRURR UR 5S95$)zDeletes a version from a model.rI)r8rQDeleter:%MlProjectsModelsVersionsDeleteRequestrSr;ris rrvVersionsClient.DeletesB ;; / / 6 6 ;;))+ < - ..rcURRRURR UR 5S95$)z-Gets details about an existing model version.ru)r8rQGetr:"MlProjectsModelsVersionsGetRequestrSrxs rr{VersionsClient.GetsB ;; / / 3 3 88))+ 9 - ..rcURRUR5S9n[R"UR R USSS9$)zLists the versions for a model.)rCversionspageSize)fieldbatch_size_attribute)r:#MlProjectsModelsVersionsListRequestrSr YieldFromListr8rQ)r;rT list_requests rListVersionsClient.ListsO==DD%%'E)L  # # ,,lz ;;rcURRRURUR 5S95$)zSets a model's default version.ru)r8rQ SetDefaultrNrSrxs rrVersionsClient.SetDefaults; ;; / / : : ##)A)A)C#D FFrc [R"U5nU(a [R"X0R5n[WR5Vs/sH+n[XVR5(dMURPM- sn5nXr- [UR55-nU(ai[SR [ R""[%U5S5SR'[)U55USR'[)U55S95eU$![Ra<n[SR U[ R "UR5S95eSnAff=fs snf)aRead a config file and return Version object with the values. The object is based on a YAML configuration file. The file may only have the fields given in `allowed_fields`. Args: path: str, the path to the YAML file. allowed_fields: Collection, the fields allowed in the YAML. Returns: A Version object (for the corresponding API version). Raises: InvalidVersionConfigFile: If the file contains unexpected fields. z:Could not read Version configuration file [{path}]: {err})patherrNzVInvalid {noun} [{fields}] in configuration file [{path}]. Allowed fields: [{allowed}].r, )nounfieldsrallowed)r load_pathErrorr formatsix text_type inner_errorr DictToMessager@set all_fieldsgetattrrIall_unrecognized_fieldsr Pluralizelenrgrh) r;rallowed_fieldsdatarr!fspecified_fieldsinvalid_fieldss r ReadConfigVersionsClient.ReadConfigs= I ^^D !d  &&t-?-?@gG,>,>,@6,@q#GVV4#AFF,@67&7'99;<=N $ ))/>>#n"5w?YYvn56ii~ 67 *0*9 :: N' JJI $ &d coo(F&G III6s#DE28E2E/37E**E/c [UR55(dgUS(doSR[UR 55VVs/sHup4U(dMSR U5PM! snn5n[ SR U55eURRUSS9Ul US(aUSURl US(aUSURl US (aNUS R 5VVs/sHupgURRXgS 9PM snnURl US (a<US Vs/sHnURRUS 9PM snURlUS (d US(a&URR!US USS9Ulggs snnfs snnfs snf)z0Adds `container` and `routes` fields to version.Nimagerz--{}z--image was not provided, but other container related flags were specified. Please specify --image or remove the following flags: {})rcommandargsenv_vars)rIraports) containerPort predict_route health_route)predicthealth)anyvaluesrgrhitemsr ValueErrorr:GoogleCloudMlV1ContainerSpecr4rrGoogleCloudMlV1EnvVarenvGoogleCloudMlV1ContainerPortrGoogleCloudMlV1RouteMapr5) r;r!kwargskv set_flagsrIraps r_ConfigureContainer"VersionsClient._ConfigureContainers v}}   '?))(.v||~(> D(>! 6== (> DFi  vi  ""  BBWoCG i"("3g f~%f~g j$J/55797kd -- - -4 - E79gg'?!"a -- 4 41 4 E"!g o&"8}}<<)'=gn#9- E9!s G "G %G#Gc 0[UR55(dgURR5UlUS(aUSURlUS(aUSURlUS(Ga/URlSUS;av[USS5nURRRURRURRRRUS95 SUS;aw[USS5nURRRURRURRRRUS95 ggg)z&Adds `auto_scaling` fields to version.N min_nodes max_nodesmetricsz cpu-usage)rItargetzgpu-duty-cycle)rrr:rcr(rXrYrintr`GoogleCloudMlV1MetricSpecNameValueValuesEnum CPU_USAGEGPU_DUTY_CYCLE)r;r!rts r_ConfigureAutoScaling$VersionsClient._ConfigureAutoScalingsa v}}   --BBDG k%+K%8g" k%+K%8g" i$&g! y) ) y!+. /##** MM 3 3]]<<PP 4   VI. . y!"23 4##** MM 3 3]]<<PP 4   /rc HU(a5URnU(dUUR-nURUU5nOUR5nUUUUUUUU U U U U S. nSn US:XaCURR 5n URR 5n!UU!lU!U lOUS:XaCURR 5n URR5n"UU"l U"U l OHUS:XaBURR 5n URR5n#UU#lU#U l U bU US'U(dURUUUUUUUUS9 U(dURUUUUS9 UR!5Hun$n%U%cM [#UU$U%5 M U$) a! Create a Version object. The object is based on an optional YAML configuration file and the parameters to this method; any provided method parameters override any provided in-file configuration. The file may only have the fields given in VersionsClientBase._ALLOWED_YAML_FIELDS specified; the only parameters allowed are those that can be specified on the command line. Args: name: str, the name of the version object to create. path: str, the path to the YAML file. deployment_uri: str, the deploymentUri to set for the Version runtime_version: str, the runtimeVersion to set for the Version labels: Version.LabelsValue, the labels to set for the version machine_type: str, the machine type to serve the model version on. description: str, the version description. framework: FrameworkValueValuesEnum, the ML framework used to train this version of the model. python_version: str, The version of Python used to train the model. prediction_class: str, the FQN of a Python class implementing the Model interface for custom prediction. package_uris: list of str, Cloud Storage URIs containing user-supplied Python code to use. accelerator_config: an accelerator config message object. service_account: Specifies the service account for resource access control. explanation_method: Enables explanations and selects the explanation method. Valid options are 'integrated-gradients' and 'sampled-shapley'. num_integral_steps: Number of integral steps for Integrated Gradients and XRAI. num_paths: Number of paths for Sampled Shapley. image: The container image to deploy. command: Entrypoint for the container image. container_args: The command-line args to pass the container. env_vars: The environment variables to set on the container. ports: The ports to which traffic will be sent in the container. predict_route: The HTTP path within the container that predict requests are sent to. health_route: The HTTP path within the container that health checks are sent to. min_nodes: The minimum number of nodes to scale this model under load. max_nodes: The maximum number of nodes to scale this model under load. metrics: List of key-value pairs to set as metrics' target for autoscaling. containers_hidden: Whether or not container-related fields are hidden on this track. autoscaling_hidden: Whether or not autoscaling fields are hidden on this track. Returns: A Version object (for the corresponding API version). Raises: InvalidVersionConfigFile: If the file contains unexpected fields. ) rIr)r2r,r-r*r+r1r0r/acceleratorConfigr3Nzintegrated-gradientszsampled-shapleyxraiexplanationConfig)rrrrrrr)rrr)_ALLOWED_YAML_FIELDS_CONTAINER_FIELDSrr@r: GoogleCloudMlV1ExplanationConfig-GoogleCloudMlV1IntegratedGradientsAttributionnumIntegralStepsintegratedGradientsAttribution(GoogleCloudMlV1SampledShapleyAttributionnumPathssampledShapleyAttributionGoogleCloudMlV1XraiAttributionxraiAttributionrrrsetattr)&r;rIrdeployment_uriruntime_versionr, machine_typer*r+python_versionprediction_classrlaccelerator_configservice_accountexplanation_methodnum_integral_steps num_pathsrrcontainer_argsrrrrrrrcontainers_hiddenautoscaling_hiddenrr!additional_fieldsexplanation_config ig_config shap_config xrai_config field_nameras& r BuildVersionVersionsClient.BuildVersionsl 00n $000n5g""$g')#"'+#/) 33==IIK--MMOi#5i :C7 0 0==IIKMMJJLk&k5@2 v %==IIKMM@@Bk%7k"+6(%/A+,   %#%    Y)W!N/446 E  U+7 Nr)r8r:)NN)NNNNNNNN)NNNNNNNNNNNNNNNNNNNNNNNNNTT)rrrrr frozensetrrr<propertyr@rFrNrRrervr{rrrrrrrrrrr&r&.s;" $  ! < 0 0( 9 ;?7;>B=A $ >7@. . ;F %N"H<"&#' $#!"&$( $&*#'&*&*!"& !% $!!%)&*9Trr&)v1)rF)r __future__rrrapitools.base.pyrrgooglecloudsdk.api_lib.utilrgooglecloudsdk.corer r googlecloudsdk.core.utilr rrr rr r$objectr&rrrrsi2&'%',*$) @z//@@Z--@/@EVEr