-zSrSSKJr SSKJr SSKJr SSKJr SSKJ r SSK J r SSK J r S r S rS rS rS rg)zGValidation of the arguments for the persistent-resources command group.)absolute_import)division)unicode_literals)util) exceptions) constants) validationcJ[R"U[RS9 g)zFValidate whether the given region is allowed for persistent resources.)available_regionsN)r ValidateRegionrSUPPORTED_TRAINING_REGIONS)regions Dlib/googlecloudsdk/command_lib/ai/persistent_resources/validation.pyr r s  D DFcLUR(a[URU5 [UR[5(a[ URU5 UR(d7[UR[5(d[ R"SS5egg)z?Validate the argument values specified in the `create` command.--resource-pool-speczNo resource pools specified. At least one resource pool must be provided via a YAML config file (--config) or via the --resource-pool-spec arg.N)resource_pool_spec_ValidateResourcePoolSpecArgs isinstance resourcePoolslist$_ValidateResourcePoolSpecsFromConfigrInvalidArgumentException)argspersistent_resource_configversions rValidateCreateArgsr!s !$"9"97C*88$??("00';  ! ! /==t D D  - - $ %% E "rc\US(d[R"SS5e[X5 g)aValidates the argument values specified via `--resource-pool-spec` flags. Args: resource_pool_specs: List[dict], a list of resource pool specs specified via command line arguments. version: str, the API version this command will interact with, either GA or BETA. rrzEEmpty value is not allowed for the first `--resource-pool-spec` flag.N)rr'_ValidateHardwareInResourcePoolSpecArgs)resource_pool_specsrs rrr2s2 Q   - -O QQ**=Grc UGHsnU(dM SU;a[R"SS5eSU;aSU;a[R"SS5eSU;aSU;a[R"SS5eSU;aSU;a[R"SS 5eS U;S U;:wa[R"SS 5eURS S 5nU(dM[R"SU5R nUR 5Vs/sHnURS5(dMUPM nnX6;dGM2[R"SSRUSRS[U555S95e g s snf)aValidates the hardware related fields specified in `--resource-pool-spec` flags. Args: resource_pool_specs: List[dict], a list of resource pool specs specified via command line arguments. version: str, the API version this command will interact with, either GA or BETA. z machine-typerz9Key [machine-type] required in dict arg but not provided.zmin-replica-countzmax-replica-countzVKey [max-replica-count] required in dict arg when key [min-replica-count] is provided.zVKey [min-replica-count] required in dict arg when key [max-replica-count] is provided.z replica-countz:Key [replica-count] required in dict arg but not provided.zaccelerator-countzaccelerator-typezPKey [accelerator-type] and [accelerator-count] are required to use accelerators.N MachineSpecNVIDIAzWFound invalid value of [accelerator-type]: {actual}. Available values are [{expected}]., c3$# UHov M g7fN.0vs r :_ValidateHardwareInResourcePoolSpecArgs..ys%E1DAa1Dactualexpected) rrgetapi_util GetMessageAcceleratorTypeValueValuesEnumnames startswithformatjoinsorted)r rspecaccelerator_type type_enumtype valid_typess rrrCs"d t t #11 " GI I  %,?t,K11 " /0 0  %,?t,K11 " /0 0  %,?t,K11 " HJ J  %+=+E F11 " ! ! "4d;  '' 7$$B$B '__. .Tx( .    .33$45;V* II%E 1D%EE6<6GH HQ"F s E<%E<c U(d[R"SS5eUGHnUR(d[R"SS5eURR(d[R"SS5eUR(d(UR (d[R"SS5eUR (a2UR R (d[R"SS5eUR (a2UR R(d[R"SS5eURR(a2URR(d[R"SS5eUR(aMURR(a2URR(d[R"SS 5eURR(dGM[URRR5n[R "S U5R"nUR%5Vs/sHnUR'S 5(dMUPM nnX6;dGMp[R"SS R)US R+S[-U555S95e gs snf)zJValidate ResourcePoolSpec message instances imported from the config file.z--configzDAt least one [resourcePools] required in but not provided in config.z;Field [machineSpec] required in but not provided in config.z;Field [machineType] required in but not provided in config.z.s#C/B!A/Br-r.N)rr machineSpec machineType replicaCountautoscalingSpecminReplicaCountmaxReplicaCountacceleratorCountacceleratorTypediskSpecbootDiskSizeGb bootDiskTypestrnamer2r3r4r5r6r7r8r9)resource_poolsrr:r;r<r=r>s rrr|sd   - -N PPd     / /  G II    ' '  / /  G II   (<(<  / /  H JJ t';';'K'K  / /  C EE t';';'K'K  / /  C EE ))    , ,  / /     }}$--66"mm88  / /     '''T--==BBC%% ""@"@%??,,4 __X & ,  ,11 2396())#Cvk/B#CC4:4EF FcXs #KKN)__doc__ __future__rrrgooglecloudsdk.api_lib.airr2googlecloudsdk.calliopergooglecloudsdk.command_lib.airr r rrrrr'rrrUs>N&'6.34F %"H"6Hr<Fr