Skip to content

Commit b54bd66

Browse files
chore: use gapic-generator-python 0.51.2 (#42)
- [ ] Regenerate this pull request now. fix: add 'dict' annotation type to 'request' Committer: @busunkim96 PiperOrigin-RevId: 398509016 Source-Link: googleapis/googleapis@b224dfa Source-Link: googleapis/googleapis-gen@63a1db7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNjNhMWRiN2EzOGQ3NGI5NjM5NTkyZjUyMWVkMWRhYWY3Mjk5YWQ5YSJ9
1 parent c67eae0 commit b54bd66

File tree

5 files changed

+42
-42
lines changed

5 files changed

+42
-42
lines changed

packages/google-cloud-tpu/google/cloud/tpu_v1/services/tpu/client.py

Lines changed: 23 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
from distutils import util
1818
import os
1919
import re
20-
from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union
20+
from typing import Dict, Optional, Sequence, Tuple, Type, Union
2121
import pkg_resources
2222

2323
from google.api_core import client_options as client_options_lib # type: ignore
@@ -391,7 +391,7 @@ def __init__(
391391

392392
def list_nodes(
393393
self,
394-
request: cloud_tpu.ListNodesRequest = None,
394+
request: Union[cloud_tpu.ListNodesRequest, dict] = None,
395395
*,
396396
parent: str = None,
397397
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -401,7 +401,7 @@ def list_nodes(
401401
r"""Lists nodes.
402402
403403
Args:
404-
request (google.cloud.tpu_v1.types.ListNodesRequest):
404+
request (Union[google.cloud.tpu_v1.types.ListNodesRequest, dict]):
405405
The request object. Request for
406406
[ListNodes][google.cloud.tpu.v1.Tpu.ListNodes].
407407
parent (str):
@@ -469,7 +469,7 @@ def list_nodes(
469469

470470
def get_node(
471471
self,
472-
request: cloud_tpu.GetNodeRequest = None,
472+
request: Union[cloud_tpu.GetNodeRequest, dict] = None,
473473
*,
474474
name: str = None,
475475
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -479,7 +479,7 @@ def get_node(
479479
r"""Gets the details of a node.
480480
481481
Args:
482-
request (google.cloud.tpu_v1.types.GetNodeRequest):
482+
request (Union[google.cloud.tpu_v1.types.GetNodeRequest, dict]):
483483
The request object. Request for
484484
[GetNode][google.cloud.tpu.v1.Tpu.GetNode].
485485
name (str):
@@ -536,7 +536,7 @@ def get_node(
536536

537537
def create_node(
538538
self,
539-
request: cloud_tpu.CreateNodeRequest = None,
539+
request: Union[cloud_tpu.CreateNodeRequest, dict] = None,
540540
*,
541541
parent: str = None,
542542
node: cloud_tpu.Node = None,
@@ -548,7 +548,7 @@ def create_node(
548548
r"""Creates a node.
549549
550550
Args:
551-
request (google.cloud.tpu_v1.types.CreateNodeRequest):
551+
request (Union[google.cloud.tpu_v1.types.CreateNodeRequest, dict]):
552552
The request object. Request for
553553
[CreateNode][google.cloud.tpu.v1.Tpu.CreateNode].
554554
parent (str):
@@ -631,7 +631,7 @@ def create_node(
631631

632632
def delete_node(
633633
self,
634-
request: cloud_tpu.DeleteNodeRequest = None,
634+
request: Union[cloud_tpu.DeleteNodeRequest, dict] = None,
635635
*,
636636
name: str = None,
637637
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -641,7 +641,7 @@ def delete_node(
641641
r"""Deletes a node.
642642
643643
Args:
644-
request (google.cloud.tpu_v1.types.DeleteNodeRequest):
644+
request (Union[google.cloud.tpu_v1.types.DeleteNodeRequest, dict]):
645645
The request object. Request for
646646
[DeleteNode][google.cloud.tpu.v1.Tpu.DeleteNode].
647647
name (str):
@@ -710,7 +710,7 @@ def delete_node(
710710

711711
def reimage_node(
712712
self,
713-
request: cloud_tpu.ReimageNodeRequest = None,
713+
request: Union[cloud_tpu.ReimageNodeRequest, dict] = None,
714714
*,
715715
retry: retries.Retry = gapic_v1.method.DEFAULT,
716716
timeout: float = None,
@@ -719,7 +719,7 @@ def reimage_node(
719719
r"""Reimages a node's OS.
720720
721721
Args:
722-
request (google.cloud.tpu_v1.types.ReimageNodeRequest):
722+
request (Union[google.cloud.tpu_v1.types.ReimageNodeRequest, dict]):
723723
The request object. Request for
724724
[ReimageNode][google.cloud.tpu.v1.Tpu.ReimageNode].
725725
retry (google.api_core.retry.Retry): Designation of what errors, if any,
@@ -770,7 +770,7 @@ def reimage_node(
770770

771771
def stop_node(
772772
self,
773-
request: cloud_tpu.StopNodeRequest = None,
773+
request: Union[cloud_tpu.StopNodeRequest, dict] = None,
774774
*,
775775
retry: retries.Retry = gapic_v1.method.DEFAULT,
776776
timeout: float = None,
@@ -779,7 +779,7 @@ def stop_node(
779779
r"""Stops a node.
780780
781781
Args:
782-
request (google.cloud.tpu_v1.types.StopNodeRequest):
782+
request (Union[google.cloud.tpu_v1.types.StopNodeRequest, dict]):
783783
The request object. Request for
784784
[StopNode][google.cloud.tpu.v1.Tpu.StopNode].
785785
retry (google.api_core.retry.Retry): Designation of what errors, if any,
@@ -830,7 +830,7 @@ def stop_node(
830830

831831
def start_node(
832832
self,
833-
request: cloud_tpu.StartNodeRequest = None,
833+
request: Union[cloud_tpu.StartNodeRequest, dict] = None,
834834
*,
835835
retry: retries.Retry = gapic_v1.method.DEFAULT,
836836
timeout: float = None,
@@ -839,7 +839,7 @@ def start_node(
839839
r"""Starts a node.
840840
841841
Args:
842-
request (google.cloud.tpu_v1.types.StartNodeRequest):
842+
request (Union[google.cloud.tpu_v1.types.StartNodeRequest, dict]):
843843
The request object. Request for
844844
[StartNode][google.cloud.tpu.v1.Tpu.StartNode].
845845
retry (google.api_core.retry.Retry): Designation of what errors, if any,
@@ -890,7 +890,7 @@ def start_node(
890890

891891
def list_tensor_flow_versions(
892892
self,
893-
request: cloud_tpu.ListTensorFlowVersionsRequest = None,
893+
request: Union[cloud_tpu.ListTensorFlowVersionsRequest, dict] = None,
894894
*,
895895
parent: str = None,
896896
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -900,7 +900,7 @@ def list_tensor_flow_versions(
900900
r"""List TensorFlow versions supported by this API.
901901
902902
Args:
903-
request (google.cloud.tpu_v1.types.ListTensorFlowVersionsRequest):
903+
request (Union[google.cloud.tpu_v1.types.ListTensorFlowVersionsRequest, dict]):
904904
The request object. Request for
905905
[ListTensorFlowVersions][google.cloud.tpu.v1.Tpu.ListTensorFlowVersions].
906906
parent (str):
@@ -970,7 +970,7 @@ def list_tensor_flow_versions(
970970

971971
def get_tensor_flow_version(
972972
self,
973-
request: cloud_tpu.GetTensorFlowVersionRequest = None,
973+
request: Union[cloud_tpu.GetTensorFlowVersionRequest, dict] = None,
974974
*,
975975
name: str = None,
976976
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -980,7 +980,7 @@ def get_tensor_flow_version(
980980
r"""Gets TensorFlow Version.
981981
982982
Args:
983-
request (google.cloud.tpu_v1.types.GetTensorFlowVersionRequest):
983+
request (Union[google.cloud.tpu_v1.types.GetTensorFlowVersionRequest, dict]):
984984
The request object. Request for
985985
[GetTensorFlowVersion][google.cloud.tpu.v1.Tpu.GetTensorFlowVersion].
986986
name (str):
@@ -1039,7 +1039,7 @@ def get_tensor_flow_version(
10391039

10401040
def list_accelerator_types(
10411041
self,
1042-
request: cloud_tpu.ListAcceleratorTypesRequest = None,
1042+
request: Union[cloud_tpu.ListAcceleratorTypesRequest, dict] = None,
10431043
*,
10441044
parent: str = None,
10451045
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -1049,7 +1049,7 @@ def list_accelerator_types(
10491049
r"""Lists accelerator types supported by this API.
10501050
10511051
Args:
1052-
request (google.cloud.tpu_v1.types.ListAcceleratorTypesRequest):
1052+
request (Union[google.cloud.tpu_v1.types.ListAcceleratorTypesRequest, dict]):
10531053
The request object. Request for
10541054
[ListAcceleratorTypes][google.cloud.tpu.v1.Tpu.ListAcceleratorTypes].
10551055
parent (str):
@@ -1117,7 +1117,7 @@ def list_accelerator_types(
11171117

11181118
def get_accelerator_type(
11191119
self,
1120-
request: cloud_tpu.GetAcceleratorTypeRequest = None,
1120+
request: Union[cloud_tpu.GetAcceleratorTypeRequest, dict] = None,
11211121
*,
11221122
name: str = None,
11231123
retry: retries.Retry = gapic_v1.method.DEFAULT,
@@ -1127,7 +1127,7 @@ def get_accelerator_type(
11271127
r"""Gets AcceleratorType.
11281128
11291129
Args:
1130-
request (google.cloud.tpu_v1.types.GetAcceleratorTypeRequest):
1130+
request (Union[google.cloud.tpu_v1.types.GetAcceleratorTypeRequest, dict]):
11311131
The request object. Request for
11321132
[GetAcceleratorType][google.cloud.tpu.v1.Tpu.GetAcceleratorType].
11331133
name (str):

packages/google-cloud-tpu/google/cloud/tpu_v1/services/tpu/transports/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def __init__(
117117
**scopes_kwargs, quota_project_id=quota_project_id
118118
)
119119

120-
# If the credentials is service account credentials, then always try to use self signed JWT.
120+
# If the credentials are service account credentials, then always try to use self signed JWT.
121121
if (
122122
always_use_jwt_access
123123
and isinstance(credentials, service_account.Credentials)

packages/google-cloud-tpu/google/cloud/tpu_v1/services/tpu/transports/grpc.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -83,16 +83,16 @@ def __init__(
8383
api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint.
8484
If provided, it overrides the ``host`` argument and tries to create
8585
a mutual TLS channel with client SSL credentials from
86-
``client_cert_source`` or applicatin default SSL credentials.
86+
``client_cert_source`` or application default SSL credentials.
8787
client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]):
8888
Deprecated. A callback to provide client SSL certificate bytes and
8989
private key bytes, both in PEM format. It is ignored if
9090
``api_mtls_endpoint`` is None.
9191
ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials
92-
for grpc channel. It is ignored if ``channel`` is provided.
92+
for the grpc channel. It is ignored if ``channel`` is provided.
9393
client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]):
9494
A callback to provide client certificate bytes and private key bytes,
95-
both in PEM format. It is used to configure mutual TLS channel. It is
95+
both in PEM format. It is used to configure a mutual TLS channel. It is
9696
ignored if ``channel`` or ``ssl_channel_credentials`` is provided.
9797
quota_project_id (Optional[str]): An optional project to use for billing
9898
and quota.

packages/google-cloud-tpu/google/cloud/tpu_v1/services/tpu/transports/grpc_asyncio.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -130,16 +130,16 @@ def __init__(
130130
api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint.
131131
If provided, it overrides the ``host`` argument and tries to create
132132
a mutual TLS channel with client SSL credentials from
133-
``client_cert_source`` or applicatin default SSL credentials.
133+
``client_cert_source`` or application default SSL credentials.
134134
client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]):
135135
Deprecated. A callback to provide client SSL certificate bytes and
136136
private key bytes, both in PEM format. It is ignored if
137137
``api_mtls_endpoint`` is None.
138138
ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials
139-
for grpc channel. It is ignored if ``channel`` is provided.
139+
for the grpc channel. It is ignored if ``channel`` is provided.
140140
client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]):
141141
A callback to provide client certificate bytes and private key bytes,
142-
both in PEM format. It is used to configure mutual TLS channel. It is
142+
both in PEM format. It is used to configure a mutual TLS channel. It is
143143
ignored if ``channel`` or ``ssl_channel_credentials`` is provided.
144144
quota_project_id (Optional[str]): An optional project to use for billing
145145
and quota.

packages/google-cloud-tpu/scripts/fixup_tpu_v1_keywords.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -39,17 +39,17 @@ def partition(
3939
class tpuCallTransformer(cst.CSTTransformer):
4040
CTRL_PARAMS: Tuple[str] = ('retry', 'timeout', 'metadata')
4141
METHOD_TO_PARAMS: Dict[str, Tuple[str]] = {
42-
'create_node': ('parent', 'node', 'node_id', ),
43-
'delete_node': ('name', ),
44-
'get_accelerator_type': ('name', ),
45-
'get_node': ('name', ),
46-
'get_tensor_flow_version': ('name', ),
47-
'list_accelerator_types': ('parent', 'page_size', 'page_token', 'filter', 'order_by', ),
48-
'list_nodes': ('parent', 'page_size', 'page_token', ),
49-
'list_tensor_flow_versions': ('parent', 'page_size', 'page_token', 'filter', 'order_by', ),
50-
'reimage_node': ('name', 'tensorflow_version', ),
51-
'start_node': ('name', ),
52-
'stop_node': ('name', ),
42+
'create_node': ('parent', 'node', 'node_id', ),
43+
'delete_node': ('name', ),
44+
'get_accelerator_type': ('name', ),
45+
'get_node': ('name', ),
46+
'get_tensor_flow_version': ('name', ),
47+
'list_accelerator_types': ('parent', 'page_size', 'page_token', 'filter', 'order_by', ),
48+
'list_nodes': ('parent', 'page_size', 'page_token', ),
49+
'list_tensor_flow_versions': ('parent', 'page_size', 'page_token', 'filter', 'order_by', ),
50+
'reimage_node': ('name', 'tensorflow_version', ),
51+
'start_node': ('name', ),
52+
'stop_node': ('name', ),
5353
}
5454

5555
def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode:
@@ -68,7 +68,7 @@ def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode:
6868
return updated
6969

7070
kwargs, ctrl_kwargs = partition(
71-
lambda a: not a.keyword.value in self.CTRL_PARAMS,
71+
lambda a: a.keyword.value not in self.CTRL_PARAMS,
7272
kwargs
7373
)
7474

0 commit comments

Comments
 (0)