|
27 | 27 | Type, |
28 | 28 | Union, |
29 | 29 | ) |
30 | | -import pkg_resources |
| 30 | + |
| 31 | +from google.cloud.automl_v1 import gapic_version as package_version |
31 | 32 |
|
32 | 33 | from google.api_core.client_options import ClientOptions |
33 | 34 | from google.api_core import exceptions as core_exceptions |
@@ -250,7 +251,7 @@ async def create_dataset( |
250 | 251 | parent: Optional[str] = None, |
251 | 252 | dataset: Optional[gca_dataset.Dataset] = None, |
252 | 253 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
253 | | - timeout: Optional[float] = None, |
| 254 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
254 | 255 | metadata: Sequence[Tuple[str, str]] = (), |
255 | 256 | ) -> operation_async.AsyncOperation: |
256 | 257 | r"""Creates a dataset. |
@@ -285,7 +286,7 @@ async def sample_create_dataset(): |
285 | 286 |
|
286 | 287 | print("Waiting for operation to complete...") |
287 | 288 |
|
288 | | - response = await operation.result() |
| 289 | + response = (await operation).result() |
289 | 290 |
|
290 | 291 | # Handle the response |
291 | 292 | print(response) |
@@ -378,7 +379,7 @@ async def get_dataset( |
378 | 379 | *, |
379 | 380 | name: Optional[str] = None, |
380 | 381 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
381 | | - timeout: Optional[float] = None, |
| 382 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
382 | 383 | metadata: Sequence[Tuple[str, str]] = (), |
383 | 384 | ) -> dataset.Dataset: |
384 | 385 | r"""Gets a dataset. |
@@ -492,7 +493,7 @@ async def list_datasets( |
492 | 493 | *, |
493 | 494 | parent: Optional[str] = None, |
494 | 495 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
495 | | - timeout: Optional[float] = None, |
| 496 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
496 | 497 | metadata: Sequence[Tuple[str, str]] = (), |
497 | 498 | ) -> pagers.ListDatasetsAsyncPager: |
498 | 499 | r"""Lists datasets in a project. |
@@ -618,7 +619,7 @@ async def update_dataset( |
618 | 619 | dataset: Optional[gca_dataset.Dataset] = None, |
619 | 620 | update_mask: Optional[field_mask_pb2.FieldMask] = None, |
620 | 621 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
621 | | - timeout: Optional[float] = None, |
| 622 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
622 | 623 | metadata: Sequence[Tuple[str, str]] = (), |
623 | 624 | ) -> gca_dataset.Dataset: |
624 | 625 | r"""Updates a dataset. |
@@ -737,7 +738,7 @@ async def delete_dataset( |
737 | 738 | *, |
738 | 739 | name: Optional[str] = None, |
739 | 740 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
740 | | - timeout: Optional[float] = None, |
| 741 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
741 | 742 | metadata: Sequence[Tuple[str, str]] = (), |
742 | 743 | ) -> operation_async.AsyncOperation: |
743 | 744 | r"""Deletes a dataset and all of its contents. Returns empty |
@@ -771,7 +772,7 @@ async def sample_delete_dataset(): |
771 | 772 |
|
772 | 773 | print("Waiting for operation to complete...") |
773 | 774 |
|
774 | | - response = await operation.result() |
| 775 | + response = (await operation).result() |
775 | 776 |
|
776 | 777 | # Handle the response |
777 | 778 | print(response) |
@@ -876,7 +877,7 @@ async def import_data( |
876 | 877 | name: Optional[str] = None, |
877 | 878 | input_config: Optional[io.InputConfig] = None, |
878 | 879 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
879 | | - timeout: Optional[float] = None, |
| 880 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
880 | 881 | metadata: Sequence[Tuple[str, str]] = (), |
881 | 882 | ) -> operation_async.AsyncOperation: |
882 | 883 | r"""Imports data into a dataset. For Tables this method can only be |
@@ -919,7 +920,7 @@ async def sample_import_data(): |
919 | 920 |
|
920 | 921 | print("Waiting for operation to complete...") |
921 | 922 |
|
922 | | - response = await operation.result() |
| 923 | + response = (await operation).result() |
923 | 924 |
|
924 | 925 | # Handle the response |
925 | 926 | print(response) |
@@ -1025,7 +1026,7 @@ async def export_data( |
1025 | 1026 | name: Optional[str] = None, |
1026 | 1027 | output_config: Optional[io.OutputConfig] = None, |
1027 | 1028 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1028 | | - timeout: Optional[float] = None, |
| 1029 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1029 | 1030 | metadata: Sequence[Tuple[str, str]] = (), |
1030 | 1031 | ) -> operation_async.AsyncOperation: |
1031 | 1032 | r"""Exports dataset's data to the provided output location. Returns |
@@ -1062,7 +1063,7 @@ async def sample_export_data(): |
1062 | 1063 |
|
1063 | 1064 | print("Waiting for operation to complete...") |
1064 | 1065 |
|
1065 | | - response = await operation.result() |
| 1066 | + response = (await operation).result() |
1066 | 1067 |
|
1067 | 1068 | # Handle the response |
1068 | 1069 | print(response) |
@@ -1165,7 +1166,7 @@ async def get_annotation_spec( |
1165 | 1166 | *, |
1166 | 1167 | name: Optional[str] = None, |
1167 | 1168 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1168 | | - timeout: Optional[float] = None, |
| 1169 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1169 | 1170 | metadata: Sequence[Tuple[str, str]] = (), |
1170 | 1171 | ) -> annotation_spec.AnnotationSpec: |
1171 | 1172 | r"""Gets an annotation spec. |
@@ -1276,7 +1277,7 @@ async def create_model( |
1276 | 1277 | parent: Optional[str] = None, |
1277 | 1278 | model: Optional[gca_model.Model] = None, |
1278 | 1279 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1279 | | - timeout: Optional[float] = None, |
| 1280 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1280 | 1281 | metadata: Sequence[Tuple[str, str]] = (), |
1281 | 1282 | ) -> operation_async.AsyncOperation: |
1282 | 1283 | r"""Creates a model. Returns a Model in the |
@@ -1310,7 +1311,7 @@ async def sample_create_model(): |
1310 | 1311 |
|
1311 | 1312 | print("Waiting for operation to complete...") |
1312 | 1313 |
|
1313 | | - response = await operation.result() |
| 1314 | + response = (await operation).result() |
1314 | 1315 |
|
1315 | 1316 | # Handle the response |
1316 | 1317 | print(response) |
@@ -1405,7 +1406,7 @@ async def get_model( |
1405 | 1406 | *, |
1406 | 1407 | name: Optional[str] = None, |
1407 | 1408 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1408 | | - timeout: Optional[float] = None, |
| 1409 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1409 | 1410 | metadata: Sequence[Tuple[str, str]] = (), |
1410 | 1411 | ) -> model.Model: |
1411 | 1412 | r"""Gets a model. |
@@ -1515,7 +1516,7 @@ async def list_models( |
1515 | 1516 | *, |
1516 | 1517 | parent: Optional[str] = None, |
1517 | 1518 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1518 | | - timeout: Optional[float] = None, |
| 1519 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1519 | 1520 | metadata: Sequence[Tuple[str, str]] = (), |
1520 | 1521 | ) -> pagers.ListModelsAsyncPager: |
1521 | 1522 | r"""Lists models. |
@@ -1640,7 +1641,7 @@ async def delete_model( |
1640 | 1641 | *, |
1641 | 1642 | name: Optional[str] = None, |
1642 | 1643 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1643 | | - timeout: Optional[float] = None, |
| 1644 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1644 | 1645 | metadata: Sequence[Tuple[str, str]] = (), |
1645 | 1646 | ) -> operation_async.AsyncOperation: |
1646 | 1647 | r"""Deletes a model. Returns ``google.protobuf.Empty`` in the |
@@ -1673,7 +1674,7 @@ async def sample_delete_model(): |
1673 | 1674 |
|
1674 | 1675 | print("Waiting for operation to complete...") |
1675 | 1676 |
|
1676 | | - response = await operation.result() |
| 1677 | + response = (await operation).result() |
1677 | 1678 |
|
1678 | 1679 | # Handle the response |
1679 | 1680 | print(response) |
@@ -1778,7 +1779,7 @@ async def update_model( |
1778 | 1779 | model: Optional[gca_model.Model] = None, |
1779 | 1780 | update_mask: Optional[field_mask_pb2.FieldMask] = None, |
1780 | 1781 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1781 | | - timeout: Optional[float] = None, |
| 1782 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1782 | 1783 | metadata: Sequence[Tuple[str, str]] = (), |
1783 | 1784 | ) -> gca_model.Model: |
1784 | 1785 | r"""Updates a model. |
@@ -1890,7 +1891,7 @@ async def deploy_model( |
1890 | 1891 | *, |
1891 | 1892 | name: Optional[str] = None, |
1892 | 1893 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
1893 | | - timeout: Optional[float] = None, |
| 1894 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
1894 | 1895 | metadata: Sequence[Tuple[str, str]] = (), |
1895 | 1896 | ) -> operation_async.AsyncOperation: |
1896 | 1897 | r"""Deploys a model. If a model is already deployed, deploying it |
@@ -1933,7 +1934,7 @@ async def sample_deploy_model(): |
1933 | 1934 |
|
1934 | 1935 | print("Waiting for operation to complete...") |
1935 | 1936 |
|
1936 | | - response = await operation.result() |
| 1937 | + response = (await operation).result() |
1937 | 1938 |
|
1938 | 1939 | # Handle the response |
1939 | 1940 | print(response) |
@@ -2027,7 +2028,7 @@ async def undeploy_model( |
2027 | 2028 | *, |
2028 | 2029 | name: Optional[str] = None, |
2029 | 2030 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
2030 | | - timeout: Optional[float] = None, |
| 2031 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
2031 | 2032 | metadata: Sequence[Tuple[str, str]] = (), |
2032 | 2033 | ) -> operation_async.AsyncOperation: |
2033 | 2034 | r"""Undeploys a model. If the model is not deployed this method has |
@@ -2065,7 +2066,7 @@ async def sample_undeploy_model(): |
2065 | 2066 |
|
2066 | 2067 | print("Waiting for operation to complete...") |
2067 | 2068 |
|
2068 | | - response = await operation.result() |
| 2069 | + response = (await operation).result() |
2069 | 2070 |
|
2070 | 2071 | # Handle the response |
2071 | 2072 | print(response) |
@@ -2160,7 +2161,7 @@ async def export_model( |
2160 | 2161 | name: Optional[str] = None, |
2161 | 2162 | output_config: Optional[io.ModelExportOutputConfig] = None, |
2162 | 2163 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
2163 | | - timeout: Optional[float] = None, |
| 2164 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
2164 | 2165 | metadata: Sequence[Tuple[str, str]] = (), |
2165 | 2166 | ) -> operation_async.AsyncOperation: |
2166 | 2167 | r"""Exports a trained, "export-able", model to a user specified |
@@ -2201,7 +2202,7 @@ async def sample_export_model(): |
2201 | 2202 |
|
2202 | 2203 | print("Waiting for operation to complete...") |
2203 | 2204 |
|
2204 | | - response = await operation.result() |
| 2205 | + response = (await operation).result() |
2205 | 2206 |
|
2206 | 2207 | # Handle the response |
2207 | 2208 | print(response) |
@@ -2306,7 +2307,7 @@ async def get_model_evaluation( |
2306 | 2307 | *, |
2307 | 2308 | name: Optional[str] = None, |
2308 | 2309 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
2309 | | - timeout: Optional[float] = None, |
| 2310 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
2310 | 2311 | metadata: Sequence[Tuple[str, str]] = (), |
2311 | 2312 | ) -> model_evaluation.ModelEvaluation: |
2312 | 2313 | r"""Gets a model evaluation. |
@@ -2417,7 +2418,7 @@ async def list_model_evaluations( |
2417 | 2418 | parent: Optional[str] = None, |
2418 | 2419 | filter: Optional[str] = None, |
2419 | 2420 | retry: OptionalRetry = gapic_v1.method.DEFAULT, |
2420 | | - timeout: Optional[float] = None, |
| 2421 | + timeout: Union[float, object] = gapic_v1.method.DEFAULT, |
2421 | 2422 | metadata: Sequence[Tuple[str, str]] = (), |
2422 | 2423 | ) -> pagers.ListModelEvaluationsAsyncPager: |
2423 | 2424 | r"""Lists model evaluations. |
@@ -2567,14 +2568,9 @@ async def __aexit__(self, exc_type, exc, tb): |
2567 | 2568 | await self.transport.close() |
2568 | 2569 |
|
2569 | 2570 |
|
2570 | | -try: |
2571 | | - DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( |
2572 | | - gapic_version=pkg_resources.get_distribution( |
2573 | | - "google-cloud-automl", |
2574 | | - ).version, |
2575 | | - ) |
2576 | | -except pkg_resources.DistributionNotFound: |
2577 | | - DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo() |
| 2571 | +DEFAULT_CLIENT_INFO = gapic_v1.client_info.ClientInfo( |
| 2572 | + gapic_version=package_version.__version__ |
| 2573 | +) |
2578 | 2574 |
|
2579 | 2575 |
|
2580 | 2576 | __all__ = ("AutoMlAsyncClient",) |
0 commit comments