vVdZddlmZddlmZddlmZGddejZ y)z,Generated client library for tpu version v2.)absolute_import)base_api)tpu_v2_messagesceZdZdZeZdZdZdZdgZ dZ dZ dZ d Z dZdZd Z dfd ZGd d ej&ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&ZGddej&ZxZS)TpuV2z4Generated client library for service tpu version v2.zhttps://tpu.googleapis.com/z https://tpu.mtls.googleapis.com/tpuz.https://www.googleapis.com/auth/cloud-platformv2 CLIENT_ID CLIENT_SECRETzgoogle-cloud-sdkNc |xs |j}tt|||||||||| | |  |j ||_|j ||_|j||_ |j||_ |j||_ |j||_|j!||_y)zCreate a new tpu handle.) credentialsget_credentialshttpmodel log_request log_responsecredentials_argsdefault_global_paramsadditional_http_headersresponse_encodingN)BASE_URLsuperr__init__(ProjectsLocationsAcceleratorTypesService#projects_locations_acceleratorTypesProjectsLocationsNodesServiceprojects_locations_nodes"ProjectsLocationsOperationsServiceprojects_locations_operations'ProjectsLocationsQueuedResourcesService"projects_locations_queuedResources'ProjectsLocationsRuntimeVersionsService"projects_locations_runtimeVersionsProjectsLocationsServiceprojects_locationsProjectsServiceprojects) selfurlr rrrrrrrrr __class__s Alib/googlecloudsdk/generated_clients/apis/tpu/v2/tpu_v2_client.pyrzTpuV2.__init__s  C % 'd%l)3 7+ -04/\/\]a/bD,$($F$Ft$LD!)-)P)PQU)VD&.2.Z.Z[_.`D+.2.Z.Z[_.`D+";;DAD((.DMcVeZdZdZdZfdZddZde_ddZde_xZ S) .TpuV2.ProjectsLocationsAcceleratorTypesServicezCService class for the projects_locations_acceleratorTypes resource.rcNttj||i|_yN)rrrr_upload_configsr(clientr*s r+rz7TpuV2.ProjectsLocationsAcceleratorTypesService.__init__7s" E : :DJ6R dr,cL|jd}|j|||S)a Gets AcceleratorType. Args: request: (TpuProjectsLocationsAcceleratorTypesGetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (AcceleratorType) The response message. Get global_paramsGetMethodConfig _RunMethodr(requestr7configs r+r5z2TpuV2.ProjectsLocationsAcceleratorTypesService.Get<1##E*f __ '88r,c Ftjddddgdggddddd S) NzVv2/projects/{projectsId}/locations/{locationsId}/acceleratorTypes/{acceleratorTypesId}GETz+tpu.projects.locations.acceleratorTypes.getname v2/{+name}.TpuProjectsLocationsAcceleratorTypesGetRequestAcceleratorTypeF flat_path http_method method_idordered_params path_params query_params relative_path request_fieldrequest_type_nameresponse_type_namesupports_downloadr ApiMethodInfor,r+z7TpuV2.ProjectsLocationsAcceleratorTypesService.Is4 6 6j?xH"J, !r,cL|jd}|j|||S)a0Lists accelerator types supported by this API. Args: request: (TpuProjectsLocationsAcceleratorTypesListRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (ListAcceleratorTypesResponse) The response message. Listr6r8r;s r+rWz3TpuV2.ProjectsLocationsAcceleratorTypesService.ListW1##F+f __ '88r,c Jtjddddgdggddddd d S) NzAv2/projects/{projectsId}/locations/{locationsId}/acceleratorTypesr@z,tpu.projects.locations.acceleratorTypes.listparentfilterorderBypageSize pageTokenzv2/{+parent}/acceleratorTypesrC/TpuProjectsLocationsAcceleratorTypesListRequestListAcceleratorTypesResponseFrFrRrTr,r+rUz7TpuV2.ProjectsLocationsAcceleratorTypesService.ds4!7!7U@ zJC5K9 "r,r0 __name__ __module__ __qualname____doc___NAMErr5 method_configrW __classcell__r*s@r+rr.2s1M 1E 8 C 8 Dr,rceZdZdZdZfdZddZde_ddZde_ddZ d e _dd Z d e _dd Z d e _ddZ de _ddZ de _ddZde_xZS)#TpuV2.ProjectsLocationsNodesServicez8Service class for the projects_locations_nodes resource.rcNttj||i|_yr0)rrrrr1r2s r+rz,TpuV2.ProjectsLocationsNodesService.__init__ws" E / /?G dr,cL|jd}|j|||S)zCreates a node. Args: request: (TpuProjectsLocationsNodesCreateRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. Creater6r8r;s r+roz*TpuV2.ProjectsLocationsNodesService.Create|1##H-f __ '88r,c Htjddddgdgdgdddd d S) N6v2/projects/{projectsId}/locations/{locationsId}/nodesPOSTz#tpu.projects.locations.nodes.createrZnodeIdv2/{+parent}/nodesnode&TpuProjectsLocationsNodesCreateRequest OperationFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.s68#9#9J7 zJZ*B& $r,cL|jd}|j|||S)zDeletes a node. Args: request: (TpuProjectsLocationsNodesDeleteRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. Deleter6r8r;s r+rzz*TpuV2.ProjectsLocationsNodesService.Deleterpr,c Ftjddddgdggddddd S) N@v2/projects/{projectsId}/locations/{locationsId}/nodes/{nodesId}DELETEz#tpu.projects.locations.nodes.deleterArBrC&TpuProjectsLocationsNodesDeleteRequestrxFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.s48#9#9T7xH"B& $r,cL|jd}|j|||S)zGets the details of a node. Args: request: (TpuProjectsLocationsNodesGetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Node) The response message. r5r6r8r;s r+r5z'TpuV2.ProjectsLocationsNodesService.Getr>r,c Ftjddddgdggddddd S) Nr|r@z tpu.projects.locations.nodes.getrArBrC#TpuProjectsLocationsNodesGetRequestNodeFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.s4 6 6T4xH"?! !r,cL|jd}|j|||S)a/Retrieves the guest attributes for the node. Args: request: (TpuProjectsLocationsNodesGetGuestAttributesRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (GetGuestAttributesResponse) The response message. GetGuestAttributesr6r8r;s r+rz6TpuV2.ProjectsLocationsNodesService.GetGuestAttributess2##$89f __ '88r,c Ftjddddgdggddddd S) NzSv2/projects/{projectsId}/locations/{locationsId}/nodes/{nodesId}:getGuestAttributesrsz/tpu.projects.locations.nodes.getGuestAttributesrAzv2/{+name}:getGuestAttributesgetGuestAttributesRequest2TpuProjectsLocationsNodesGetGuestAttributesRequestGetGuestAttributesResponseFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.s4x/E/EgCxH51N7 0r,cL|jd}|j|||S)zLists nodes. Args: request: (TpuProjectsLocationsNodesListRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (ListNodesResponse) The response message. rWr6r8r;s r+rWz(TpuV2.ProjectsLocationsNodesService.ListrXr,c Jtjddddgdgddgddd d d S) Nrrr@z!tpu.projects.locations.nodes.listrZr^r_rurC$TpuProjectsLocationsNodesListRequestListNodesResponseFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.s9!7!7J5 zJ +.*@. "r,cL|jd}|j|||S)a Updates the configurations of a node. Args: request: (TpuProjectsLocationsNodesPatchRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. Patchr6r8r;s r+rz)TpuV2.ProjectsLocationsNodesService.Patch1##G,f __ '88r,c Htjddddgdgdgdddd d S) Nr|PATCHz"tpu.projects.locations.nodes.patchrA updateMaskrBrv%TpuProjectsLocationsNodesPatchRequestrxFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.s6("8"8T6xH"^"A& #r,cL|jd}|j|||S)zStarts a node. Args: request: (TpuProjectsLocationsNodesStartRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. Startr6r8r;s r+rz)TpuV2.ProjectsLocationsNodesService.Startrr,c Ftjddddgdggddddd S) NzFv2/projects/{projectsId}/locations/{locationsId}/nodes/{nodesId}:startrsz"tpu.projects.locations.nodes.startrAzv2/{+name}:startstartNodeRequest%TpuProjectsLocationsNodesStartRequestrxFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.+s4("8"8Z6xH((A& #r,cL|jd}|j|||S)a)Stops a node. This operation is only available with single TPU nodes. Args: request: (TpuProjectsLocationsNodesStopRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. Stopr6r8r;s r+rz(TpuV2.ProjectsLocationsNodesService.Stop9rXr,c Ftjddddgdggddddd S) NzEv2/projects/{projectsId}/locations/{locationsId}/nodes/{nodesId}:stoprsz!tpu.projects.locations.nodes.stoprAzv2/{+name}:stopstopNodeRequest$TpuProjectsLocationsNodesStopRequestrxFrFrRrTr,r+rUz,TpuV2.ProjectsLocationsNodesService.Fs4!7!7Y5xH''@& "r,r0)rcrdrerfrgrrorhrzr5rrWrrrrirjs@r+rrlrsB &E 8 F 8 F 8 C 8 ($ 8 D 8 E 8 E 8 Dr,rceZdZdZdZfdZd dZde_d dZde_d dZ d e _d d Z d e _xZ S) (TpuV2.ProjectsLocationsOperationsServicez=Service class for the projects_locations_operations resource.rcNttj||i|_yr0)rrrrr1r2s r+rz1TpuV2.ProjectsLocationsOperationsService.__init__Ys" E 4 4dDVL dr,cL|jd}|j|||S)a2Starts asynchronous cancellation on a long-running operation. The server makes a best effort to cancel the operation, but success is not guaranteed. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Clients can use Operations.GetOperation or other methods to check whether the cancellation succeeded or whether the operation completed despite cancellation. On successful cancellation, the operation is not deleted; instead, it becomes an operation with an Operation.error value with a google.rpc.Status.code of `1`, corresponding to `Code.CANCELLED`. Args: request: (TpuProjectsLocationsOperationsCancelRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Empty) The response message. Cancelr6r8r;s r+rz/TpuV2.ProjectsLocationsOperationsService.Cancel^rpr,c Ftjddddgdggddddd S) NzQv2/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}:cancelrsz(tpu.projects.locations.operations.cancelrAzv2/{+name}:cancelrC+TpuProjectsLocationsOperationsCancelRequestEmptyFrFrRrTr,r+rUz1TpuV2.ProjectsLocationsOperationsService.ks48#9#9e<xH)G" $r,cL|jd}|j|||S)aDeletes a long-running operation. This method indicates that the client is no longer interested in the operation result. It does not cancel the operation. If the server doesn't support this method, it returns `google.rpc.Code.UNIMPLEMENTED`. Args: request: (TpuProjectsLocationsOperationsDeleteRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Empty) The response message. rzr6r8r;s r+rzz/TpuV2.ProjectsLocationsOperationsService.Deleteyrpr,c Ftjddddgdggddddd S) NJv2/projects/{projectsId}/locations/{locationsId}/operations/{operationsId}r}z(tpu.projects.locations.operations.deleterArBrC+TpuProjectsLocationsOperationsDeleteRequestrFrFrRrTr,r+rUz1TpuV2.ProjectsLocationsOperationsService.s48#9#9^<xH"G" $r,cL|jd}|j|||S)aGets the latest state of a long-running operation. Clients can use this method to poll the operation result at intervals as recommended by the API service. Args: request: (TpuProjectsLocationsOperationsGetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. r5r6r8r;s r+r5z,TpuV2.ProjectsLocationsOperationsService.Getr>r,c Ftjddddgdggddddd S) Nrr@z%tpu.projects.locations.operations.getrArBrC(TpuProjectsLocationsOperationsGetRequestrxFrFrRrTr,r+rUz1TpuV2.ProjectsLocationsOperationsService.s4 6 6^9xH"D& !r,cL|jd}|j|||S)a}Lists operations that match the specified filter in the request. If the server doesn't support this method, it returns `UNIMPLEMENTED`. Args: request: (TpuProjectsLocationsOperationsListRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (ListOperationsResponse) The response message. rWr6r8r;s r+rWz-TpuV2.ProjectsLocationsOperationsService.ListrXr,c Jtjddddgdggddddd d S) Nz;v2/projects/{projectsId}/locations/{locationsId}/operationsr@z&tpu.projects.locations.operations.listrA)r\r^r_zv2/{+name}/operationsrC)TpuProjectsLocationsOperationsListRequestListOperationsResponseFrFrRrTr,r+rUz1TpuV2.ProjectsLocationsOperationsService.s4!7!7O:xH8-E3 "r,r0) rcrdrerfrgrrrhrzr5rWrirjs@r+rrTsOG +E 8 F 8 F 8 C 8 Dr,rceZdZdZdZfdZddZde_ddZde_ddZ d e _dd Z d e _dd Z d e _xZ S)-TpuV2.ProjectsLocationsQueuedResourcesServicezBService class for the projects_locations_queuedResources resource.r!cNttj||i|_yr0)rrr rr1r2s r+rz6TpuV2.ProjectsLocationsQueuedResourcesService.__init__" E 9 94I&Q dr,cL|jd}|j|||S)aCreates a QueuedResource TPU instance. Args: request: (TpuProjectsLocationsQueuedResourcesCreateRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. ror6r8r;s r+roz4TpuV2.ProjectsLocationsQueuedResourcesService.Createrpr,c Jtjddddgdgddgddd d d S) N@v2/projects/{projectsId}/locations/{locationsId}/queuedResourcesrsz-tpu.projects.locations.queuedResources.createrZqueuedResourceId requestIdv2/{+parent}/queuedResourcesqueuedResource0TpuProjectsLocationsQueuedResourcesCreateRequestrxFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsQueuedResourcesService.s98#9#9TA zJ(+64&L& $r,cL|jd}|j|||S)aDeletes a QueuedResource TPU instance. Args: request: (TpuProjectsLocationsQueuedResourcesDeleteRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. rzr6r8r;s r+rzz4TpuV2.ProjectsLocationsQueuedResourcesService.Deleterpr,c Jtjddddgdgddgddd d d S) NTv2/projects/{projectsId}/locations/{locationsId}/queuedResources/{queuedResourcesId}r}z-tpu.projects.locations.queuedResources.deleterAforcerrBrC0TpuProjectsLocationsQueuedResourcesDeleteRequestrxFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsQueuedResourcesService.s98#9#9hAxH{+"L& $r,cL|jd}|j|||S)aGets details of a queued resource. Args: request: (TpuProjectsLocationsQueuedResourcesGetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (QueuedResource) The response message. r5r6r8r;s r+r5z1TpuV2.ProjectsLocationsQueuedResourcesService.Get r>r,c Ftjddddgdggddddd S) Nrr@z*tpu.projects.locations.queuedResources.getrArBrC-TpuProjectsLocationsQueuedResourcesGetRequestQueuedResourceFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsQueuedResourcesService.4 6 6h>xH"I+ !r,cL|jd}|j|||S)aLists queued resources. Args: request: (TpuProjectsLocationsQueuedResourcesListRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (ListQueuedResourcesResponse) The response message. rWr6r8r;s r+rWz2TpuV2.ProjectsLocationsQueuedResourcesService.List%rXr,c Jtjddddgdgddgddd d d S) Nrr@z+tpu.projects.locations.queuedResources.listrZr^r_rrC.TpuProjectsLocationsQueuedResourcesListRequestListQueuedResourcesResponseFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsQueuedResourcesService.2s9!7!7T? zJ +.4J8 "r,cL|jd}|j|||S)aResets a QueuedResource TPU instance. Args: request: (TpuProjectsLocationsQueuedResourcesResetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Operation) The response message. Resetr6r8r;s r+rz3TpuV2.ProjectsLocationsQueuedResourcesService.Reset@rr,c Ftjddddgdggddddd S) NzZv2/projects/{projectsId}/locations/{locationsId}/queuedResources/{queuedResourcesId}:resetrsz,tpu.projects.locations.queuedResources.resetrAzv2/{+name}:resetresetQueuedResourceRequest/TpuProjectsLocationsQueuedResourcesResetRequestrxFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsQueuedResourcesService.Ms4("8"8n@xH(2K& #r,r0) rcrdrerfrgrrorhrzr5rWrrirjs@r+r rs^L 0E 8 F 8 F 8 C 8 D 8 Er,r cVeZdZdZdZfdZddZde_ddZde_xZ S) -TpuV2.ProjectsLocationsRuntimeVersionsServicezBService class for the projects_locations_runtimeVersions resource.r#cNttj||i|_yr0)rrr"rr1r2s r+rz6TpuV2.ProjectsLocationsRuntimeVersionsService.__init__`rr,cL|jd}|j|||S)a Gets a runtime version. Args: request: (TpuProjectsLocationsRuntimeVersionsGetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (RuntimeVersion) The response message. r5r6r8r;s r+r5z1TpuV2.ProjectsLocationsRuntimeVersionsService.Geter>r,c Ftjddddgdggddddd S) NzTv2/projects/{projectsId}/locations/{locationsId}/runtimeVersions/{runtimeVersionsId}r@z*tpu.projects.locations.runtimeVersions.getrArBrC-TpuProjectsLocationsRuntimeVersionsGetRequestRuntimeVersionFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsRuntimeVersionsService.rrr,cL|jd}|j|||S)a-Lists runtime versions supported by this API. Args: request: (TpuProjectsLocationsRuntimeVersionsListRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (ListRuntimeVersionsResponse) The response message. rWr6r8r;s r+rWz2TpuV2.ProjectsLocationsRuntimeVersionsService.ListrXr,c Jtjddddgdggddddd d S) Nz@v2/projects/{projectsId}/locations/{locationsId}/runtimeVersionsr@z+tpu.projects.locations.runtimeVersions.listrZr[zv2/{+parent}/runtimeVersionsrC.TpuProjectsLocationsRuntimeVersionsListRequestListRuntimeVersionsResponseFrFrRrTr,r+rUz6TpuV2.ProjectsLocationsRuntimeVersionsService.s4!7!7T? zJC4J8 "r,r0rbrjs@r+r"r[s1L 0E 8 C 8 Dr,r"cneZdZdZdZfdZd dZde_d dZde_d dZ d e _xZ S) TpuV2.ProjectsLocationsServicez2Service class for the projects_locations resource.r%cNttj||i|_yr0)rrr$rr1r2s r+rz'TpuV2.ProjectsLocationsService.__init__s" E * *D:6B dr,cL|jd}|j|||S)aAGenerates the Cloud TPU service identity for the project. Args: request: (TpuProjectsLocationsGenerateServiceIdentityRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (GenerateServiceIdentityResponse) The response message. GenerateServiceIdentityr6r8r;s r+rz6TpuV2.ProjectsLocationsService.GenerateServiceIdentitys2##$=>f __ '88r,c Ftjddddgdggddddd S) NzHv2/projects/{projectsId}/locations/{locationsId}:generateServiceIdentityrsz.tpu.projects.locations.generateServiceIdentityrZz$v2/{+parent}:generateServiceIdentitygenerateServiceIdentityRequest2TpuProjectsLocationsGenerateServiceIdentityRequestGenerateServiceIdentityResponseFrFrRrTr,r+rUz'TpuV2.ProjectsLocationsService.s4H4J4J\B zJ<6N< 5r,cL|jd}|j|||S)zGets information about a location. Args: request: (TpuProjectsLocationsGetRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (Location) The response message. r5r6r8r;s r+r5z"TpuV2.ProjectsLocationsService.Getr>r,c Ftjddddgdggddddd S) Nz0v2/projects/{projectsId}/locations/{locationsId}r@ztpu.projects.locations.getrArBrCTpuProjectsLocationsGetRequestLocationFrFrRrTr,r+rUz'TpuV2.ProjectsLocationsService.s4 6 6D.xH":% !r,cL|jd}|j|||S)a,Lists information about the supported locations for this service. Args: request: (TpuProjectsLocationsListRequest) input message global_params: (StandardQueryParameters, default: None) global arguments Returns: (ListLocationsResponse) The response message. rWr6r8r;s r+rWz#TpuV2.ProjectsLocationsService.ListrXr,c Jtjddddgdggddddd d S) Nz"v2/projects/{projectsId}/locationsr@ztpu.projects.locations.listrA)extraLocationTypesr\r^r_zv2/{+name}/locationsrCTpuProjectsLocationsListRequestListLocationsResponseFrFrRrTr,r+rUz'TpuV2.ProjectsLocationsService.s4!7!76/xHN,;2 "r,r0) rcrdrerfrgrrrhr5rWrirjs@r+r$rsA< E 8 -) 8 C 8 Dr,r$c&eZdZdZdZfdZxZS)TpuV2.ProjectsServicez(Service class for the projects resource.r'cNttj||i|_yr0)rrr&rr1r2s r+rzTpuV2.ProjectsService.__init__s" E ! !41&9 dr,)rcrdrerfrgrrirjs@r+r&rs2 E  r,r&) rCNTNNFFNNNN)rcrdrerfmessagesMESSAGES_MODULEr MTLS_BASE_URL_PACKAGE_SCOPES_VERSION _CLIENT_ID_CLIENT_SECRET _USER_AGENT_CLIENT_CLASS_NAME _URL_VERSION_API_KEYrrBaseApiServicerrrr r"r$r&rirjs@r+rr s</ *(4- ( = >' (*"."+, ()-6:/4<@?C /.>1H1H>@`h&=&=`Dt8+B+BtlO0G0GOb>0G0G>@Y!8!8Yv //  r,rN) rf __future__rapitools.base.pyr,googlecloudsdk.generated_clients.apis.tpu.v2rr BaseApiClientrrTr,r+rs'2'%Tt  H " "t  r,