TpuAsyncClient(*, credentials: Optional[google.auth.credentials.Credentials] = None, transport: Union[str, google.cloud.tpu_v2alpha1.services.tpu.transports.base.TpuTransport] = 'grpc_asyncio', client_options: Optional[google.api_core.client_options.ClientOptions] = None, client_info: google.api_core.gapic_v1.client_info.ClientInfo = <google.api_core.gapic_v1.client_info.ClientInfo object>)
Manages TPU nodes and other resources TPU API v2alpha1
Inheritance
builtins.object > TpuAsyncClientProperties
transport
Returns the transport used by the client instance.
Type | Description |
TpuTransport | The transport used by the client instance. |
Methods
TpuAsyncClient
TpuAsyncClient(*, credentials: Optional[google.auth.credentials.Credentials] = None, transport: Union[str, google.cloud.tpu_v2alpha1.services.tpu.transports.base.TpuTransport] = 'grpc_asyncio', client_options: Optional[google.api_core.client_options.ClientOptions] = None, client_info: google.api_core.gapic_v1.client_info.ClientInfo = <google.api_core.gapic_v1.client_info.ClientInfo object>)
Instantiates the tpu client.
Name | Description |
credentials |
Optional[google.auth.credentials.Credentials]
The authorization credentials to attach to requests. These credentials identify the application to the service; if none are specified, the client will attempt to ascertain the credentials from the environment. |
transport |
Union[str, `.TpuTransport`]
The transport to use. If set to None, a transport is chosen automatically. |
client_options |
ClientOptions
Custom options for the client. It won't take effect if a |
Type | Description |
google.auth.exceptions.MutualTlsChannelError | If mutual TLS transport creation failed for any reason. |
accelerator_type_path
accelerator_type_path(project: str, location: str, accelerator_type: str)
Returns a fully-qualified accelerator_type string.
common_billing_account_path
common_billing_account_path(billing_account: str)
Returns a fully-qualified billing_account string.
common_folder_path
common_folder_path(folder: str)
Returns a fully-qualified folder string.
common_location_path
common_location_path(project: str, location: str)
Returns a fully-qualified location string.
common_organization_path
common_organization_path(organization: str)
Returns a fully-qualified organization string.
common_project_path
common_project_path(project: str)
Returns a fully-qualified project string.
create_node
create_node(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.CreateNodeRequest, dict]] = None, *, parent: Optional[str] = None, node: Optional[google.cloud.tpu_v2alpha1.types.cloud_tpu.Node] = None, node_id: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Creates a node.
from google.cloud import tpu_v2alpha1
async def sample_create_node():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
node = tpu_v2alpha1.Node()
node.accelerator_type = "accelerator_type_value"
node.runtime_version = "runtime_version_value"
request = tpu_v2alpha1.CreateNodeRequest(
parent="parent_value",
node=node,
)
# Make the request
operation = client.create_node(request=request)
print("Waiting for operation to complete...")
response = await operation.result()
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.CreateNodeRequest, dict]
The request object. Request for CreateNode. |
parent |
`str`
Required. The parent resource name. This corresponds to the |
node |
Node
Required. The node. This corresponds to the |
node_id |
`str`
The unqualified resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.api_core.operation_async.AsyncOperation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
delete_node
delete_node(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.DeleteNodeRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Deletes a node.
from google.cloud import tpu_v2alpha1
async def sample_delete_node():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.DeleteNodeRequest(
name="name_value",
)
# Make the request
operation = client.delete_node(request=request)
print("Waiting for operation to complete...")
response = await operation.result()
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.DeleteNodeRequest, dict]
The request object. Request for DeleteNode. |
name |
`str`
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.api_core.operation_async.AsyncOperation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
from_service_account_file
from_service_account_file(filename: str, *args, **kwargs)
Creates an instance of this client using the provided credentials file.
Name | Description |
filename |
str
The path to the service account private key json file. |
Type | Description |
TpuAsyncClient | The constructed client. |
from_service_account_info
from_service_account_info(info: dict, *args, **kwargs)
Creates an instance of this client using the provided credentials info.
Name | Description |
info |
dict
The service account private key info. |
Type | Description |
TpuAsyncClient | The constructed client. |
from_service_account_json
from_service_account_json(filename: str, *args, **kwargs)
Creates an instance of this client using the provided credentials file.
Name | Description |
filename |
str
The path to the service account private key json file. |
Type | Description |
TpuAsyncClient | The constructed client. |
generate_service_identity
generate_service_identity(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.GenerateServiceIdentityRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Generates the Cloud TPU service identity for the project.
from google.cloud import tpu_v2alpha1
async def sample_generate_service_identity():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.GenerateServiceIdentityRequest(
parent="parent_value",
)
# Make the request
response = await client.generate_service_identity(request=request)
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.GenerateServiceIdentityRequest, dict]
The request object. Request for GenerateServiceIdentity. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.types.GenerateServiceIdentityResponse | Response for GenerateServiceIdentity. |
get_accelerator_type
get_accelerator_type(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.GetAcceleratorTypeRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Gets AcceleratorType.
from google.cloud import tpu_v2alpha1
async def sample_get_accelerator_type():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.GetAcceleratorTypeRequest(
name="name_value",
)
# Make the request
response = await client.get_accelerator_type(request=request)
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.GetAcceleratorTypeRequest, dict]
The request object. Request for GetAcceleratorType. |
name |
`str`
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.types.AcceleratorType | A accelerator type that a Node can be configured with. |
get_guest_attributes
get_guest_attributes(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.GetGuestAttributesRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Retrieves the guest attributes for the node.
from google.cloud import tpu_v2alpha1
async def sample_get_guest_attributes():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.GetGuestAttributesRequest(
name="name_value",
)
# Make the request
response = await client.get_guest_attributes(request=request)
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.GetGuestAttributesRequest, dict]
The request object. Request for GetGuestAttributes. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.types.GetGuestAttributesResponse | Response for GetGuestAttributes. |
get_mtls_endpoint_and_cert_source
get_mtls_endpoint_and_cert_source(
client_options: Optional[google.api_core.client_options.ClientOptions] = None,
)
Return the API endpoint and client cert source for mutual TLS.
The client cert source is determined in the following order:
(1) if GOOGLE_API_USE_CLIENT_CERTIFICATE
environment variable is not "true", the
client cert source is None.
(2) if client_options.client_cert_source
is provided, use the provided one; if the
default client cert source exists, use the default one; otherwise the client cert
source is None.
The API endpoint is determined in the following order:
(1) if client_options.api_endpoint
if provided, use the provided one.
(2) if GOOGLE_API_USE_CLIENT_CERTIFICATE
environment variable is "always", use the
default mTLS endpoint; if the environment variabel is "never", use the default API
endpoint; otherwise if client cert source exists, use the default mTLS endpoint, otherwise
use the default API endpoint.
More details can be found at https://google.aip.dev/auth/4114.
Name | Description |
client_options |
google.api_core.client_options.ClientOptions
Custom options for the client. Only the |
Type | Description |
google.auth.exceptions.MutualTLSChannelError | If any errors happen. |
Type | Description |
Tuple[str, Callable[[], Tuple[bytes, bytes]]] | returns the API endpoint and the client cert source to use. |
get_node
get_node(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.GetNodeRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Gets the details of a node.
from google.cloud import tpu_v2alpha1
async def sample_get_node():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.GetNodeRequest(
name="name_value",
)
# Make the request
response = await client.get_node(request=request)
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.GetNodeRequest, dict]
The request object. Request for GetNode. |
name |
`str`
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.types.Node | A TPU instance. |
get_runtime_version
get_runtime_version(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.GetRuntimeVersionRequest, dict]] = None, *, name: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Gets a runtime version.
from google.cloud import tpu_v2alpha1
async def sample_get_runtime_version():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.GetRuntimeVersionRequest(
name="name_value",
)
# Make the request
response = await client.get_runtime_version(request=request)
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.GetRuntimeVersionRequest, dict]
The request object. Request for GetRuntimeVersion. |
name |
`str`
Required. The resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.types.RuntimeVersion | A runtime version that a Node can be configured with. |
get_transport_class
get_transport_class()
Returns an appropriate transport class.
list_accelerator_types
list_accelerator_types(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.ListAcceleratorTypesRequest, dict]] = None, *, parent: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Lists accelerator types supported by this API.
from google.cloud import tpu_v2alpha1
async def sample_list_accelerator_types():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.ListAcceleratorTypesRequest(
parent="parent_value",
)
# Make the request
page_result = client.list_accelerator_types(request=request)
# Handle the response
async for response in page_result:
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.ListAcceleratorTypesRequest, dict]
The request object. Request for ListAcceleratorTypes. |
parent |
`str`
Required. The parent resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.services.tpu.pagers.ListAcceleratorTypesAsyncPager | Response for ListAcceleratorTypes. Iterating over this object will yield results and resolve additional pages automatically. |
list_nodes
list_nodes(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.ListNodesRequest, dict]] = None, *, parent: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Lists nodes.
from google.cloud import tpu_v2alpha1
async def sample_list_nodes():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.ListNodesRequest(
parent="parent_value",
)
# Make the request
page_result = client.list_nodes(request=request)
# Handle the response
async for response in page_result:
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.ListNodesRequest, dict]
The request object. Request for ListNodes. |
parent |
`str`
Required. The parent resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.services.tpu.pagers.ListNodesAsyncPager | Response for ListNodes. Iterating over this object will yield results and resolve additional pages automatically. |
list_runtime_versions
list_runtime_versions(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.ListRuntimeVersionsRequest, dict]] = None, *, parent: Optional[str] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Lists runtime versions supported by this API.
from google.cloud import tpu_v2alpha1
async def sample_list_runtime_versions():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.ListRuntimeVersionsRequest(
parent="parent_value",
)
# Make the request
page_result = client.list_runtime_versions(request=request)
# Handle the response
async for response in page_result:
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.ListRuntimeVersionsRequest, dict]
The request object. Request for ListRuntimeVersions. |
parent |
`str`
Required. The parent resource name. This corresponds to the |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.cloud.tpu_v2alpha1.services.tpu.pagers.ListRuntimeVersionsAsyncPager | Response for ListRuntimeVersions. Iterating over this object will yield results and resolve additional pages automatically. |
node_path
node_path(project: str, location: str, node: str)
Returns a fully-qualified node string.
parse_accelerator_type_path
parse_accelerator_type_path(path: str)
Parses a accelerator_type path into its component segments.
parse_common_billing_account_path
parse_common_billing_account_path(path: str)
Parse a billing_account path into its component segments.
parse_common_folder_path
parse_common_folder_path(path: str)
Parse a folder path into its component segments.
parse_common_location_path
parse_common_location_path(path: str)
Parse a location path into its component segments.
parse_common_organization_path
parse_common_organization_path(path: str)
Parse a organization path into its component segments.
parse_common_project_path
parse_common_project_path(path: str)
Parse a project path into its component segments.
parse_node_path
parse_node_path(path: str)
Parses a node path into its component segments.
parse_runtime_version_path
parse_runtime_version_path(path: str)
Parses a runtime_version path into its component segments.
runtime_version_path
runtime_version_path(project: str, location: str, runtime_version: str)
Returns a fully-qualified runtime_version string.
start_node
start_node(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.StartNodeRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Starts a node.
from google.cloud import tpu_v2alpha1
async def sample_start_node():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.StartNodeRequest(
)
# Make the request
operation = client.start_node(request=request)
print("Waiting for operation to complete...")
response = await operation.result()
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.StartNodeRequest, dict]
The request object. Request for StartNode. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.api_core.operation_async.AsyncOperation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
stop_node
stop_node(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.StopNodeRequest, dict]] = None, *, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Stops a node. This operation is only available with single TPU nodes.
from google.cloud import tpu_v2alpha1
async def sample_stop_node():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
request = tpu_v2alpha1.StopNodeRequest(
)
# Make the request
operation = client.stop_node(request=request)
print("Waiting for operation to complete...")
response = await operation.result()
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.StopNodeRequest, dict]
The request object. Request for StopNode. |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.api_core.operation_async.AsyncOperation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |
update_node
update_node(request: Optional[Union[google.cloud.tpu_v2alpha1.types.cloud_tpu.UpdateNodeRequest, dict]] = None, *, node: Optional[google.cloud.tpu_v2alpha1.types.cloud_tpu.Node] = None, update_mask: Optional[google.protobuf.field_mask_pb2.FieldMask] = None, retry: Union[google.api_core.retry.Retry, google.api_core.gapic_v1.method._MethodDefault] = <_MethodDefault._DEFAULT_VALUE: <object object>>, timeout: Optional[float] = None, metadata: Sequence[Tuple[str, str]] = ())
Updates the configurations of a node.
from google.cloud import tpu_v2alpha1
async def sample_update_node():
# Create a client
client = tpu_v2alpha1.TpuAsyncClient()
# Initialize request argument(s)
node = tpu_v2alpha1.Node()
node.accelerator_type = "accelerator_type_value"
node.runtime_version = "runtime_version_value"
request = tpu_v2alpha1.UpdateNodeRequest(
node=node,
)
# Make the request
operation = client.update_node(request=request)
print("Waiting for operation to complete...")
response = await operation.result()
# Handle the response
print(response)
Name | Description |
request |
Union[google.cloud.tpu_v2alpha1.types.UpdateNodeRequest, dict]
The request object. Request for UpdateNode. |
node |
Node
Required. The node. Only fields specified in update_mask are updated. This corresponds to the |
update_mask |
`google.protobuf.field_mask_pb2.FieldMask`
Required. Mask of fields from |
retry |
google.api_core.retry.Retry
Designation of what errors, if any, should be retried. |
timeout |
float
The timeout for this request. |
metadata |
Sequence[Tuple[str, str]]
Strings which should be sent along with the request as metadata. |
Type | Description |
google.api_core.operation_async.AsyncOperation | An object representing a long-running operation. The result type for the operation will be Node A TPU instance. |