|
16 | 16 | from collections import OrderedDict |
17 | 17 | import functools |
18 | 18 | import re |
19 | | -from typing import Dict, Mapping, Optional, Sequence, Tuple, Type, Union |
| 19 | +from typing import ( |
| 20 | + Dict, |
| 21 | + Mapping, |
| 22 | + MutableMapping, |
| 23 | + MutableSequence, |
| 24 | + Optional, |
| 25 | + Sequence, |
| 26 | + Tuple, |
| 27 | + Type, |
| 28 | + Union, |
| 29 | +) |
20 | 30 |
|
21 | 31 | from google.api_core import exceptions as core_exceptions |
22 | 32 | from google.api_core import gapic_v1 |
@@ -161,9 +171,9 @@ def transport(self) -> TpuTransport: |
161 | 171 | def __init__( |
162 | 172 | self, |
163 | 173 | *, |
164 | | - credentials: ga_credentials.Credentials = None, |
| 174 | + credentials: Optional[ga_credentials.Credentials] = None, |
165 | 175 | transport: Union[str, TpuTransport] = "grpc_asyncio", |
166 | | - client_options: ClientOptions = None, |
| 176 | + client_options: Optional[ClientOptions] = None, |
167 | 177 | client_info: gapic_v1.client_info.ClientInfo = DEFAULT_CLIENT_INFO, |
168 | 178 | ) -> None: |
169 | 179 | """Instantiates the tpu client. |
@@ -207,11 +217,11 @@ def __init__( |
207 | 217 |
|
208 | 218 | async def list_nodes( |
209 | 219 | self, |
210 | | - request: Union[cloud_tpu.ListNodesRequest, dict] = None, |
| 220 | + request: Optional[Union[cloud_tpu.ListNodesRequest, dict]] = None, |
211 | 221 | *, |
212 | | - parent: str = None, |
| 222 | + parent: Optional[str] = None, |
213 | 223 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
214 | | - timeout: float = None, |
| 224 | + timeout: Optional[float] = None, |
215 | 225 | metadata: Sequence[Tuple[str, str]] = (), |
216 | 226 | ) -> pagers.ListNodesAsyncPager: |
217 | 227 | r"""Lists nodes. |
@@ -244,7 +254,7 @@ async def sample_list_nodes(): |
244 | 254 | print(response) |
245 | 255 |
|
246 | 256 | Args: |
247 | | - request (Union[google.cloud.tpu_v1.types.ListNodesRequest, dict]): |
| 257 | + request (Optional[Union[google.cloud.tpu_v1.types.ListNodesRequest, dict]]): |
248 | 258 | The request object. Request for |
249 | 259 | [ListNodes][google.cloud.tpu.v1.Tpu.ListNodes]. |
250 | 260 | parent (:class:`str`): |
@@ -320,11 +330,11 @@ async def sample_list_nodes(): |
320 | 330 |
|
321 | 331 | async def get_node( |
322 | 332 | self, |
323 | | - request: Union[cloud_tpu.GetNodeRequest, dict] = None, |
| 333 | + request: Optional[Union[cloud_tpu.GetNodeRequest, dict]] = None, |
324 | 334 | *, |
325 | | - name: str = None, |
| 335 | + name: Optional[str] = None, |
326 | 336 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
327 | | - timeout: float = None, |
| 337 | + timeout: Optional[float] = None, |
328 | 338 | metadata: Sequence[Tuple[str, str]] = (), |
329 | 339 | ) -> cloud_tpu.Node: |
330 | 340 | r"""Gets the details of a node. |
@@ -356,7 +366,7 @@ async def sample_get_node(): |
356 | 366 | print(response) |
357 | 367 |
|
358 | 368 | Args: |
359 | | - request (Union[google.cloud.tpu_v1.types.GetNodeRequest, dict]): |
| 369 | + request (Optional[Union[google.cloud.tpu_v1.types.GetNodeRequest, dict]]): |
360 | 370 | The request object. Request for |
361 | 371 | [GetNode][google.cloud.tpu.v1.Tpu.GetNode]. |
362 | 372 | name (:class:`str`): |
@@ -418,13 +428,13 @@ async def sample_get_node(): |
418 | 428 |
|
419 | 429 | async def create_node( |
420 | 430 | self, |
421 | | - request: Union[cloud_tpu.CreateNodeRequest, dict] = None, |
| 431 | + request: Optional[Union[cloud_tpu.CreateNodeRequest, dict]] = None, |
422 | 432 | *, |
423 | | - parent: str = None, |
424 | | - node: cloud_tpu.Node = None, |
425 | | - node_id: str = None, |
| 433 | + parent: Optional[str] = None, |
| 434 | + node: Optional[cloud_tpu.Node] = None, |
| 435 | + node_id: Optional[str] = None, |
426 | 436 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
427 | | - timeout: float = None, |
| 437 | + timeout: Optional[float] = None, |
428 | 438 | metadata: Sequence[Tuple[str, str]] = (), |
429 | 439 | ) -> operation_async.AsyncOperation: |
430 | 440 | r"""Creates a node. |
@@ -465,7 +475,7 @@ async def sample_create_node(): |
465 | 475 | print(response) |
466 | 476 |
|
467 | 477 | Args: |
468 | | - request (Union[google.cloud.tpu_v1.types.CreateNodeRequest, dict]): |
| 478 | + request (Optional[Union[google.cloud.tpu_v1.types.CreateNodeRequest, dict]]): |
469 | 479 | The request object. Request for |
470 | 480 | [CreateNode][google.cloud.tpu.v1.Tpu.CreateNode]. |
471 | 481 | parent (:class:`str`): |
@@ -553,11 +563,11 @@ async def sample_create_node(): |
553 | 563 |
|
554 | 564 | async def delete_node( |
555 | 565 | self, |
556 | | - request: Union[cloud_tpu.DeleteNodeRequest, dict] = None, |
| 566 | + request: Optional[Union[cloud_tpu.DeleteNodeRequest, dict]] = None, |
557 | 567 | *, |
558 | | - name: str = None, |
| 568 | + name: Optional[str] = None, |
559 | 569 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
560 | | - timeout: float = None, |
| 570 | + timeout: Optional[float] = None, |
561 | 571 | metadata: Sequence[Tuple[str, str]] = (), |
562 | 572 | ) -> operation_async.AsyncOperation: |
563 | 573 | r"""Deletes a node. |
@@ -593,7 +603,7 @@ async def sample_delete_node(): |
593 | 603 | print(response) |
594 | 604 |
|
595 | 605 | Args: |
596 | | - request (Union[google.cloud.tpu_v1.types.DeleteNodeRequest, dict]): |
| 606 | + request (Optional[Union[google.cloud.tpu_v1.types.DeleteNodeRequest, dict]]): |
597 | 607 | The request object. Request for |
598 | 608 | [DeleteNode][google.cloud.tpu.v1.Tpu.DeleteNode]. |
599 | 609 | name (:class:`str`): |
@@ -667,10 +677,10 @@ async def sample_delete_node(): |
667 | 677 |
|
668 | 678 | async def reimage_node( |
669 | 679 | self, |
670 | | - request: Union[cloud_tpu.ReimageNodeRequest, dict] = None, |
| 680 | + request: Optional[Union[cloud_tpu.ReimageNodeRequest, dict]] = None, |
671 | 681 | *, |
672 | 682 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
673 | | - timeout: float = None, |
| 683 | + timeout: Optional[float] = None, |
674 | 684 | metadata: Sequence[Tuple[str, str]] = (), |
675 | 685 | ) -> operation_async.AsyncOperation: |
676 | 686 | r"""Reimages a node's OS. |
@@ -705,7 +715,7 @@ async def sample_reimage_node(): |
705 | 715 | print(response) |
706 | 716 |
|
707 | 717 | Args: |
708 | | - request (Union[google.cloud.tpu_v1.types.ReimageNodeRequest, dict]): |
| 718 | + request (Optional[Union[google.cloud.tpu_v1.types.ReimageNodeRequest, dict]]): |
709 | 719 | The request object. Request for |
710 | 720 | [ReimageNode][google.cloud.tpu.v1.Tpu.ReimageNode]. |
711 | 721 | retry (google.api_core.retry.Retry): Designation of what errors, if any, |
@@ -760,10 +770,10 @@ async def sample_reimage_node(): |
760 | 770 |
|
761 | 771 | async def stop_node( |
762 | 772 | self, |
763 | | - request: Union[cloud_tpu.StopNodeRequest, dict] = None, |
| 773 | + request: Optional[Union[cloud_tpu.StopNodeRequest, dict]] = None, |
764 | 774 | *, |
765 | 775 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
766 | | - timeout: float = None, |
| 776 | + timeout: Optional[float] = None, |
767 | 777 | metadata: Sequence[Tuple[str, str]] = (), |
768 | 778 | ) -> operation_async.AsyncOperation: |
769 | 779 | r"""Stops a node. |
@@ -798,7 +808,7 @@ async def sample_stop_node(): |
798 | 808 | print(response) |
799 | 809 |
|
800 | 810 | Args: |
801 | | - request (Union[google.cloud.tpu_v1.types.StopNodeRequest, dict]): |
| 811 | + request (Optional[Union[google.cloud.tpu_v1.types.StopNodeRequest, dict]]): |
802 | 812 | The request object. Request for |
803 | 813 | [StopNode][google.cloud.tpu.v1.Tpu.StopNode]. |
804 | 814 | retry (google.api_core.retry.Retry): Designation of what errors, if any, |
@@ -853,10 +863,10 @@ async def sample_stop_node(): |
853 | 863 |
|
854 | 864 | async def start_node( |
855 | 865 | self, |
856 | | - request: Union[cloud_tpu.StartNodeRequest, dict] = None, |
| 866 | + request: Optional[Union[cloud_tpu.StartNodeRequest, dict]] = None, |
857 | 867 | *, |
858 | 868 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
859 | | - timeout: float = None, |
| 869 | + timeout: Optional[float] = None, |
860 | 870 | metadata: Sequence[Tuple[str, str]] = (), |
861 | 871 | ) -> operation_async.AsyncOperation: |
862 | 872 | r"""Starts a node. |
@@ -891,7 +901,7 @@ async def sample_start_node(): |
891 | 901 | print(response) |
892 | 902 |
|
893 | 903 | Args: |
894 | | - request (Union[google.cloud.tpu_v1.types.StartNodeRequest, dict]): |
| 904 | + request (Optional[Union[google.cloud.tpu_v1.types.StartNodeRequest, dict]]): |
895 | 905 | The request object. Request for |
896 | 906 | [StartNode][google.cloud.tpu.v1.Tpu.StartNode]. |
897 | 907 | retry (google.api_core.retry.Retry): Designation of what errors, if any, |
@@ -946,11 +956,11 @@ async def sample_start_node(): |
946 | 956 |
|
947 | 957 | async def list_tensor_flow_versions( |
948 | 958 | self, |
949 | | - request: Union[cloud_tpu.ListTensorFlowVersionsRequest, dict] = None, |
| 959 | + request: Optional[Union[cloud_tpu.ListTensorFlowVersionsRequest, dict]] = None, |
950 | 960 | *, |
951 | | - parent: str = None, |
| 961 | + parent: Optional[str] = None, |
952 | 962 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
953 | | - timeout: float = None, |
| 963 | + timeout: Optional[float] = None, |
954 | 964 | metadata: Sequence[Tuple[str, str]] = (), |
955 | 965 | ) -> pagers.ListTensorFlowVersionsAsyncPager: |
956 | 966 | r"""List TensorFlow versions supported by this API. |
@@ -983,7 +993,7 @@ async def sample_list_tensor_flow_versions(): |
983 | 993 | print(response) |
984 | 994 |
|
985 | 995 | Args: |
986 | | - request (Union[google.cloud.tpu_v1.types.ListTensorFlowVersionsRequest, dict]): |
| 996 | + request (Optional[Union[google.cloud.tpu_v1.types.ListTensorFlowVersionsRequest, dict]]): |
987 | 997 | The request object. Request for |
988 | 998 | [ListTensorFlowVersions][google.cloud.tpu.v1.Tpu.ListTensorFlowVersions]. |
989 | 999 | parent (:class:`str`): |
@@ -1059,11 +1069,11 @@ async def sample_list_tensor_flow_versions(): |
1059 | 1069 |
|
1060 | 1070 | async def get_tensor_flow_version( |
1061 | 1071 | self, |
1062 | | - request: Union[cloud_tpu.GetTensorFlowVersionRequest, dict] = None, |
| 1072 | + request: Optional[Union[cloud_tpu.GetTensorFlowVersionRequest, dict]] = None, |
1063 | 1073 | *, |
1064 | | - name: str = None, |
| 1074 | + name: Optional[str] = None, |
1065 | 1075 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1066 | | - timeout: float = None, |
| 1076 | + timeout: Optional[float] = None, |
1067 | 1077 | metadata: Sequence[Tuple[str, str]] = (), |
1068 | 1078 | ) -> cloud_tpu.TensorFlowVersion: |
1069 | 1079 | r"""Gets TensorFlow Version. |
@@ -1095,7 +1105,7 @@ async def sample_get_tensor_flow_version(): |
1095 | 1105 | print(response) |
1096 | 1106 |
|
1097 | 1107 | Args: |
1098 | | - request (Union[google.cloud.tpu_v1.types.GetTensorFlowVersionRequest, dict]): |
| 1108 | + request (Optional[Union[google.cloud.tpu_v1.types.GetTensorFlowVersionRequest, dict]]): |
1099 | 1109 | The request object. Request for |
1100 | 1110 | [GetTensorFlowVersion][google.cloud.tpu.v1.Tpu.GetTensorFlowVersion]. |
1101 | 1111 | name (:class:`str`): |
@@ -1159,11 +1169,11 @@ async def sample_get_tensor_flow_version(): |
1159 | 1169 |
|
1160 | 1170 | async def list_accelerator_types( |
1161 | 1171 | self, |
1162 | | - request: Union[cloud_tpu.ListAcceleratorTypesRequest, dict] = None, |
| 1172 | + request: Optional[Union[cloud_tpu.ListAcceleratorTypesRequest, dict]] = None, |
1163 | 1173 | *, |
1164 | | - parent: str = None, |
| 1174 | + parent: Optional[str] = None, |
1165 | 1175 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1166 | | - timeout: float = None, |
| 1176 | + timeout: Optional[float] = None, |
1167 | 1177 | metadata: Sequence[Tuple[str, str]] = (), |
1168 | 1178 | ) -> pagers.ListAcceleratorTypesAsyncPager: |
1169 | 1179 | r"""Lists accelerator types supported by this API. |
@@ -1196,7 +1206,7 @@ async def sample_list_accelerator_types(): |
1196 | 1206 | print(response) |
1197 | 1207 |
|
1198 | 1208 | Args: |
1199 | | - request (Union[google.cloud.tpu_v1.types.ListAcceleratorTypesRequest, dict]): |
| 1209 | + request (Optional[Union[google.cloud.tpu_v1.types.ListAcceleratorTypesRequest, dict]]): |
1200 | 1210 | The request object. Request for |
1201 | 1211 | [ListAcceleratorTypes][google.cloud.tpu.v1.Tpu.ListAcceleratorTypes]. |
1202 | 1212 | parent (:class:`str`): |
@@ -1272,11 +1282,11 @@ async def sample_list_accelerator_types(): |
1272 | 1282 |
|
1273 | 1283 | async def get_accelerator_type( |
1274 | 1284 | self, |
1275 | | - request: Union[cloud_tpu.GetAcceleratorTypeRequest, dict] = None, |
| 1285 | + request: Optional[Union[cloud_tpu.GetAcceleratorTypeRequest, dict]] = None, |
1276 | 1286 | *, |
1277 | | - name: str = None, |
| 1287 | + name: Optional[str] = None, |
1278 | 1288 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1279 | | - timeout: float = None, |
| 1289 | + timeout: Optional[float] = None, |
1280 | 1290 | metadata: Sequence[Tuple[str, str]] = (), |
1281 | 1291 | ) -> cloud_tpu.AcceleratorType: |
1282 | 1292 | r"""Gets AcceleratorType. |
@@ -1308,7 +1318,7 @@ async def sample_get_accelerator_type(): |
1308 | 1318 | print(response) |
1309 | 1319 |
|
1310 | 1320 | Args: |
1311 | | - request (Union[google.cloud.tpu_v1.types.GetAcceleratorTypeRequest, dict]): |
| 1321 | + request (Optional[Union[google.cloud.tpu_v1.types.GetAcceleratorTypeRequest, dict]]): |
1312 | 1322 | The request object. Request for |
1313 | 1323 | [GetAcceleratorType][google.cloud.tpu.v1.Tpu.GetAcceleratorType]. |
1314 | 1324 | name (:class:`str`): |
|
0 commit comments