8SrSSKJr SSKJr SSKJr SSKrSSKJr SSK J r SSK J r SS K J r SS KJr SS KJr SS KJr S rSrSrSrSrSrSSjrSrSrSrSrSr Sr!Sr"Sr#g)z:Validations of the arguments of custom-jobs command group.)absolute_import)division)unicode_literalsN)util) exceptions) constants) validation) local_util)utils)filescJ[R"U[RS9 g)zIValidate whether the given region is allowed for specifically custom job.)available_regionsN)r ValidateRegionrSUPPORTED_TRAINING_REGIONS)regions ;lib/googlecloudsdk/command_lib/ai/custom_jobs/validation.pyrr s  D DFcjUR(a[URU5 g[U5 g)z;Validate the argument values specified in `create` command.N)worker_pool_spec_ValidateWorkerPoolSpecArgs"_ValidateWorkerPoolSpecsFromConfig)argsjob_spec_from_configversions rValidateCreateArgsr&s&  5 5w?&';8_ValidateHardwareInWorkerPoolSpecArgs..es%E1DAa1Ds)actualexpected) rrgetapi_util GetMessageAcceleratorTypeValueValuesEnumnames startswithformatjoinsorted)r! api_versionspecaccelerator_type type_enumtype valid_typess rrrAs0 d t t #11 GI I  $);4)G11 R  "4d;  '' ;((F(F '__. .Tx((DOOE,B .    .33"45;V* II%E 1D%EE6<6GH H- " s 0D(D(c^[US5n[U5S:a[USSU5 gg)zJValidates the software fields specified in all `--worker-pool-spec` flags.rN)&_ValidateSoftwareInFirstWorkerPoolSpeclen&_ValidateSoftwareInRestWorkerPoolSpecs)r!has_local_packages rr r hs?< a*+ rc>SU;a [U5 g[U5 g)zValidates the software related fields specified in the first `--worker-pool-spec` flags. Args: spec: dict, the specification of the first worker pool. Returns: A boolean value whether a local package will be used. local-package-pathTF)+_ValidateWorkerPoolSoftwareWithLocalPackage/_ValidateWorkerPoolSoftwareWithoutLocalPackages)r9s rr@r@rs"T!/5 3D9 rcUHhnU(dM U(a-1Skn[URUR55SS9 M@SU;a[R"SS5e[ U5 Mj g)a&Validates the argument values specified in all but the first `--worker-pool-spec` flags. Args: specs: List[dict], the list all but the first worker pool specs specified in command line. is_local_package_specified: bool, whether local package is specified in the first worker pool. >script requirements extra-dirs python-moduleextra-packagesexecutor-image-uricontainer-image-urizA local package has been specified in the first `--worker-pool-spec` flag and to be used for all workers, do not specify these keys elsewhere.unexpected_keysreasonrErzPKey [local-package-path] is only allowed in the first `--worker-pool-spec` flag.N)_RaiseErrorIfUnexpectedKeys intersectionkeysrrrG)specsis_local_package_specifiedr9software_fieldss rrBrBsod t #  $+88E; = 4 '33",. . 8=1rcSU;de[USSS9 SU;a[R"SS5eSU;SU;-S:wa[R"SS 5eS U;agUS nURS 5S n/S QnX#;a-UR S5(d[R"SS5e[ R "U5 gg![a+n[R"SSRX55eSnAff=f)z^Validate the software in a single `--worker-pool-spec` when `local-package-path` is specified.rEr) flag_namerNzLKey [executor-image-uri] is required when `local-package-path` is specified.rLrIr?z_Exactly one of keys [python-module, script] is required when `local-package-path` is specified.output-image-uri/r)zgcr.ioz eu.gcr.ioz asia.gcr.ioz us.gcr.ioz-docker.pkg.devzSThe value of `output-image-uri` has to be a valid gcr.io or Artifact Registry image+'{}' is not a valid container image uri: {}N) _RaiseErrorIfNotExistsrrsplitendswith docker_utilsValidateRepositoryAndTag ValueErrorr5)r9 output_imagehostnamecontainer_registrieses rrFrFs1  %% %  ,@B%  - -V  (d"23q8  - - 2 33 4*+L!!#&q)HN+H4E4E55  / /  _  ++L9    / /  8 ? ?    s/C C<&C77C<c^SU;deSU;nSU;nSU;nX-S:wa[R"SS5eU(aU(a[R"SS5eU(aU(d[R"SS 5e1S knURUR55n[ US S 9 g )zbValidate the software in a single `--worker-pool-spec` when `local-package-path` is not specified.rErNrOrLr?rzJExactly one of keys [executor-image-uri, container-image-uri] is required.zKKey [python-module] is not allowed together with key [container-image-uri].z Key [python-module] is required.>rIrJrKrMz_Only allow to specify together with `local-package-path` in the first `--worker-pool-spec` flag)rRN)rrrTrUrS)r9has_executor_imagehas_container_imagehas_python_modulelocal_package_only_keysrQs rrGrGs T )) )+t3-5%-.14  - -   .  - - " $$  1  - -@ BB ,88E/ nrc U(a=[R"SSRSR[ U55US95eg)Nrz(Keys [{keys}] are not allowed: {reason}.r&)rUrR)rrr5r6r7rPs rrSrSsJ  - -HOO6/23FPD EErcrURHnUR=(a3 URR=(d URRnUR=(a URR nU(aU(dU(aMU(aM[ R"SS5e g)zHValidate WorkerPoolSpec message instances imported from the config file.z--configzuExactly one of fields [pythonPackageSpec, containerSpec] is required for a [workerPoolSpecs] in the YAML config file.N)workerPoolSpecspythonPackageSpecexecutorImageUri pythonModule containerSpecimageUrirr)job_specr9use_python_package use_containers rrrs&&d//- // , ++&&F4+=+=+F+FM,mm6H6H  / /  J LL'rc/nUR(aURS5 UR(aURS5 UR(aURS5 UR(aURS5 UR (aURS5 UR (aURS5 U$)zAReturns names of all the flags specified only for image building.rIrLrJrMrKr[)rIappend python_modulerJextra_packages extra_dirsoutput_image_uri)rimage_build_argss r_ImageBuildArgSpecifiedr s [[H% O, N+ ,- __L) ./ rc [U5nU(dgUR(dQUR(d@[R"SS/SR SR [U5555eUR(aSnO'[R"UR5UlSn[RR[RR URUR55n[RRU5(a$[RRU5(d;[R "USR URUR55eUR"=(d /Hn[RR[RR URU55n[RRU5(a&[RRU5(aM[R "SSR X@R55e UR$=(d /Hn[RR[RR URU55n[RRU5(a&[RR'U5(aM[R "S S R X`R55e UR((a$UR(n[*R,"U5 g[*R0"URS 9Ulg![.a+n [R "S S R X55eSn A ff=f)zEValidates the arguments related to image building and normalize them.Nz--scriptz--python-modulezThey are required to build a training container image. Otherwise, please remove flags [{}] to directly run the `executor-image-uri`.r&z/File '{}' is not found under the package: '{}'.z--extra-packagesz7Package file '{}' is not found under the package: '{}'.z --extra-dirsz4Directory '{}' is not found under the package: '{}'.z--output-image-urir]) base_name)rrIrzrMinimumArgumentExceptionr5r6r7r ModuleToPathospathnormpathlocal_package_pathexistsisfilerr{r|isdirr}rarbrcGenerateImageName) rbuild_args_specifiedarg_name script_pathpackage package_path directorydir_pathrdrgs r_ValidBuildArgsOfLocalRunrs06  T//  - - &' X  &!567 8  :: [[H))$*<*<=DK H  ggll4**DKK8:+  $ $BGGNN;,G,G  - -DKK KK002 33 %%++g77##  T,,g68L 77>>, ' 'rww~~l/K/K  / /  D K K.. 0 11 ,OO)r)iww  T,,i8:H 77>>( # #277==+B+B  / /  A H H00 2 33 * ((L ++L9)::++D   / /  8 ? ?    sN O &OO cURnU(a[RR[R "U55n[RR U5(a$[RRU5(d&[R"SSRU55eO[R"5nX l[U5 U$)zLValidates the arguments specified in `local-run` command and normalize them.z--local-package-pathDirectory '{}' is not found.) rrrabspathr ExpandHomeDirrrrrr5GetCWDr)rargs_local_package_pachwork_dirs rValidateLocalRunArgsr\s 33wwu223JKLH 77>>( # #277==+B+B  / / ) 0 0 : <<,C ||~H$D! +rcF[RR[R"U55n[RR U5(a$[RR U5(d&[R"USRU55eg)zValidate the local package is valid. Args: local_package_path: str, path of the local directory to check. flag_name: str, indicates in which flag the path is specified. rN) rrrr rrrrrr5)rrZrs rr^r^nssWW__U001CD E(  ! !x)@)@  - -299(C EE*Ar)F)$__doc__ __future__rrrrgooglecloudsdk.api_lib.airr0googlecloudsdk.calliopergooglecloudsdk.command_lib.airr )googlecloudsdk.command_lib.ai.custom_jobsr $googlecloudsdk.command_lib.ai.dockerr ragooglecloudsdk.core.utilr rrrrr r@rBrFrGrSrrrrr^r(rrrsA&' 6.34@F*F =;$$HN>$GL">J" J#LEL"&;|$ Er