diff --git a/google/cloud/automl_v1/services/auto_ml/client.py b/google/cloud/automl_v1/services/auto_ml/client.py index bc732357..c82a6566 100644 --- a/google/cloud/automl_v1/services/auto_ml/client.py +++ b/google/cloud/automl_v1/services/auto_ml/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources from google.api_core import client_options as client_options_lib # type: ignore @@ -443,7 +443,7 @@ def __init__( def create_dataset( self, - request: service.CreateDatasetRequest = None, + request: Union[service.CreateDatasetRequest, dict] = None, *, parent: str = None, dataset: gca_dataset.Dataset = None, @@ -454,7 +454,7 @@ def create_dataset( r"""Creates a dataset. Args: - request (google.cloud.automl_v1.types.CreateDatasetRequest): + request (Union[google.cloud.automl_v1.types.CreateDatasetRequest, dict]): The request object. Request message for [AutoMl.CreateDataset][google.cloud.automl.v1.AutoMl.CreateDataset]. parent (str): @@ -532,7 +532,7 @@ def create_dataset( def get_dataset( self, - request: service.GetDatasetRequest = None, + request: Union[service.GetDatasetRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -542,7 +542,7 @@ def get_dataset( r"""Gets a dataset. Args: - request (google.cloud.automl_v1.types.GetDatasetRequest): + request (Union[google.cloud.automl_v1.types.GetDatasetRequest, dict]): The request object. Request message for [AutoMl.GetDataset][google.cloud.automl.v1.AutoMl.GetDataset]. name (str): @@ -605,7 +605,7 @@ def get_dataset( def list_datasets( self, - request: service.ListDatasetsRequest = None, + request: Union[service.ListDatasetsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -615,7 +615,7 @@ def list_datasets( r"""Lists datasets in a project. Args: - request (google.cloud.automl_v1.types.ListDatasetsRequest): + request (Union[google.cloud.automl_v1.types.ListDatasetsRequest, dict]): The request object. Request message for [AutoMl.ListDatasets][google.cloud.automl.v1.AutoMl.ListDatasets]. parent (str): @@ -685,7 +685,7 @@ def list_datasets( def update_dataset( self, - request: service.UpdateDatasetRequest = None, + request: Union[service.UpdateDatasetRequest, dict] = None, *, dataset: gca_dataset.Dataset = None, update_mask: field_mask_pb2.FieldMask = None, @@ -696,7 +696,7 @@ def update_dataset( r"""Updates a dataset. Args: - request (google.cloud.automl_v1.types.UpdateDatasetRequest): + request (Union[google.cloud.automl_v1.types.UpdateDatasetRequest, dict]): The request object. Request message for [AutoMl.UpdateDataset][google.cloud.automl.v1.AutoMl.UpdateDataset] dataset (google.cloud.automl_v1.types.Dataset): @@ -770,7 +770,7 @@ def update_dataset( def delete_dataset( self, - request: service.DeleteDatasetRequest = None, + request: Union[service.DeleteDatasetRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -784,7 +784,7 @@ def delete_dataset( [metadata][google.longrunning.Operation.metadata] field. Args: - request (google.cloud.automl_v1.types.DeleteDatasetRequest): + request (Union[google.cloud.automl_v1.types.DeleteDatasetRequest, dict]): The request object. Request message for [AutoMl.DeleteDataset][google.cloud.automl.v1.AutoMl.DeleteDataset]. name (str): @@ -866,7 +866,7 @@ def delete_dataset( def import_data( self, - request: service.ImportDataRequest = None, + request: Union[service.ImportDataRequest, dict] = None, *, name: str = None, input_config: io.InputConfig = None, @@ -886,7 +886,7 @@ def import_data( field when it completes. Args: - request (google.cloud.automl_v1.types.ImportDataRequest): + request (Union[google.cloud.automl_v1.types.ImportDataRequest, dict]): The request object. Request message for [AutoMl.ImportData][google.cloud.automl.v1.AutoMl.ImportData]. name (str): @@ -979,7 +979,7 @@ def import_data( def export_data( self, - request: service.ExportDataRequest = None, + request: Union[service.ExportDataRequest, dict] = None, *, name: str = None, output_config: io.OutputConfig = None, @@ -993,7 +993,7 @@ def export_data( completes. Args: - request (google.cloud.automl_v1.types.ExportDataRequest): + request (Union[google.cloud.automl_v1.types.ExportDataRequest, dict]): The request object. Request message for [AutoMl.ExportData][google.cloud.automl.v1.AutoMl.ExportData]. name (str): @@ -1084,7 +1084,7 @@ def export_data( def get_annotation_spec( self, - request: service.GetAnnotationSpecRequest = None, + request: Union[service.GetAnnotationSpecRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1094,7 +1094,7 @@ def get_annotation_spec( r"""Gets an annotation spec. Args: - request (google.cloud.automl_v1.types.GetAnnotationSpecRequest): + request (Union[google.cloud.automl_v1.types.GetAnnotationSpecRequest, dict]): The request object. Request message for [AutoMl.GetAnnotationSpec][google.cloud.automl.v1.AutoMl.GetAnnotationSpec]. name (str): @@ -1153,7 +1153,7 @@ def get_annotation_spec( def create_model( self, - request: service.CreateModelRequest = None, + request: Union[service.CreateModelRequest, dict] = None, *, parent: str = None, model: gca_model.Model = None, @@ -1168,7 +1168,7 @@ def create_model( each annotation spec. Args: - request (google.cloud.automl_v1.types.CreateModelRequest): + request (Union[google.cloud.automl_v1.types.CreateModelRequest, dict]): The request object. Request message for [AutoMl.CreateModel][google.cloud.automl.v1.AutoMl.CreateModel]. parent (str): @@ -1248,7 +1248,7 @@ def create_model( def get_model( self, - request: service.GetModelRequest = None, + request: Union[service.GetModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1258,7 +1258,7 @@ def get_model( r"""Gets a model. Args: - request (google.cloud.automl_v1.types.GetModelRequest): + request (Union[google.cloud.automl_v1.types.GetModelRequest, dict]): The request object. Request message for [AutoMl.GetModel][google.cloud.automl.v1.AutoMl.GetModel]. name (str): @@ -1317,7 +1317,7 @@ def get_model( def list_models( self, - request: service.ListModelsRequest = None, + request: Union[service.ListModelsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1327,7 +1327,7 @@ def list_models( r"""Lists models. Args: - request (google.cloud.automl_v1.types.ListModelsRequest): + request (Union[google.cloud.automl_v1.types.ListModelsRequest, dict]): The request object. Request message for [AutoMl.ListModels][google.cloud.automl.v1.AutoMl.ListModels]. parent (str): @@ -1397,7 +1397,7 @@ def list_models( def delete_model( self, - request: service.DeleteModelRequest = None, + request: Union[service.DeleteModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1410,7 +1410,7 @@ def delete_model( [metadata][google.longrunning.Operation.metadata] field. Args: - request (google.cloud.automl_v1.types.DeleteModelRequest): + request (Union[google.cloud.automl_v1.types.DeleteModelRequest, dict]): The request object. Request message for [AutoMl.DeleteModel][google.cloud.automl.v1.AutoMl.DeleteModel]. name (str): @@ -1492,7 +1492,7 @@ def delete_model( def update_model( self, - request: service.UpdateModelRequest = None, + request: Union[service.UpdateModelRequest, dict] = None, *, model: gca_model.Model = None, update_mask: field_mask_pb2.FieldMask = None, @@ -1503,7 +1503,7 @@ def update_model( r"""Updates a model. Args: - request (google.cloud.automl_v1.types.UpdateModelRequest): + request (Union[google.cloud.automl_v1.types.UpdateModelRequest, dict]): The request object. Request message for [AutoMl.UpdateModel][google.cloud.automl.v1.AutoMl.UpdateModel] model (google.cloud.automl_v1.types.Model): @@ -1575,7 +1575,7 @@ def update_model( def deploy_model( self, - request: service.DeployModelRequest = None, + request: Union[service.DeployModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1599,7 +1599,7 @@ def deploy_model( completes. Args: - request (google.cloud.automl_v1.types.DeployModelRequest): + request (Union[google.cloud.automl_v1.types.DeployModelRequest, dict]): The request object. Request message for [AutoMl.DeployModel][google.cloud.automl.v1.AutoMl.DeployModel]. name (str): @@ -1681,7 +1681,7 @@ def deploy_model( def undeploy_model( self, - request: service.UndeployModelRequest = None, + request: Union[service.UndeployModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1699,7 +1699,7 @@ def undeploy_model( completes. Args: - request (google.cloud.automl_v1.types.UndeployModelRequest): + request (Union[google.cloud.automl_v1.types.UndeployModelRequest, dict]): The request object. Request message for [AutoMl.UndeployModel][google.cloud.automl.v1.AutoMl.UndeployModel]. name (str): @@ -1781,7 +1781,7 @@ def undeploy_model( def export_model( self, - request: service.ExportModelRequest = None, + request: Union[service.ExportModelRequest, dict] = None, *, name: str = None, output_config: io.ModelExportOutputConfig = None, @@ -1799,7 +1799,7 @@ def export_model( completes. Args: - request (google.cloud.automl_v1.types.ExportModelRequest): + request (Union[google.cloud.automl_v1.types.ExportModelRequest, dict]): The request object. Request message for [AutoMl.ExportModel][google.cloud.automl.v1.AutoMl.ExportModel]. Models need to be enabled for exporting, otherwise an @@ -1892,7 +1892,7 @@ def export_model( def get_model_evaluation( self, - request: service.GetModelEvaluationRequest = None, + request: Union[service.GetModelEvaluationRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1902,7 +1902,7 @@ def get_model_evaluation( r"""Gets a model evaluation. Args: - request (google.cloud.automl_v1.types.GetModelEvaluationRequest): + request (Union[google.cloud.automl_v1.types.GetModelEvaluationRequest, dict]): The request object. Request message for [AutoMl.GetModelEvaluation][google.cloud.automl.v1.AutoMl.GetModelEvaluation]. name (str): @@ -1961,7 +1961,7 @@ def get_model_evaluation( def list_model_evaluations( self, - request: service.ListModelEvaluationsRequest = None, + request: Union[service.ListModelEvaluationsRequest, dict] = None, *, parent: str = None, filter: str = None, @@ -1972,7 +1972,7 @@ def list_model_evaluations( r"""Lists model evaluations. Args: - request (google.cloud.automl_v1.types.ListModelEvaluationsRequest): + request (Union[google.cloud.automl_v1.types.ListModelEvaluationsRequest, dict]): The request object. Request message for [AutoMl.ListModelEvaluations][google.cloud.automl.v1.AutoMl.ListModelEvaluations]. parent (str): diff --git a/google/cloud/automl_v1/services/auto_ml/transports/base.py b/google/cloud/automl_v1/services/auto_ml/transports/base.py index 89ade31a..36e216d8 100644 --- a/google/cloud/automl_v1/services/auto_ml/transports/base.py +++ b/google/cloud/automl_v1/services/auto_ml/transports/base.py @@ -123,7 +123,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/automl_v1/services/auto_ml/transports/grpc.py b/google/cloud/automl_v1/services/auto_ml/transports/grpc.py index dd01c16b..e6aa5422 100644 --- a/google/cloud/automl_v1/services/auto_ml/transports/grpc.py +++ b/google/cloud/automl_v1/services/auto_ml/transports/grpc.py @@ -102,16 +102,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1/services/auto_ml/transports/grpc_asyncio.py b/google/cloud/automl_v1/services/auto_ml/transports/grpc_asyncio.py index 32bf2047..f137ea5b 100644 --- a/google/cloud/automl_v1/services/auto_ml/transports/grpc_asyncio.py +++ b/google/cloud/automl_v1/services/auto_ml/transports/grpc_asyncio.py @@ -149,16 +149,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1/services/prediction_service/client.py b/google/cloud/automl_v1/services/prediction_service/client.py index d64407c4..a2539dd3 100644 --- a/google/cloud/automl_v1/services/prediction_service/client.py +++ b/google/cloud/automl_v1/services/prediction_service/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources from google.api_core import client_options as client_options_lib # type: ignore @@ -363,7 +363,7 @@ def __init__( def predict( self, - request: prediction_service.PredictRequest = None, + request: Union[prediction_service.PredictRequest, dict] = None, *, name: str = None, payload: data_items.ExamplePayload = None, @@ -411,7 +411,7 @@ def predict( up to 5MB. Not available for FORECASTING ``prediction_type``. Args: - request (google.cloud.automl_v1.types.PredictRequest): + request (Union[google.cloud.automl_v1.types.PredictRequest, dict]): The request object. Request message for [PredictionService.Predict][google.cloud.automl.v1.PredictionService.Predict]. name (str): @@ -520,7 +520,7 @@ def predict( def batch_predict( self, - request: prediction_service.BatchPredictRequest = None, + request: Union[prediction_service.BatchPredictRequest, dict] = None, *, name: str = None, input_config: io.BatchPredictInputConfig = None, @@ -552,7 +552,7 @@ def batch_predict( - AutoML Tables Args: - request (google.cloud.automl_v1.types.BatchPredictRequest): + request (Union[google.cloud.automl_v1.types.BatchPredictRequest, dict]): The request object. Request message for [PredictionService.BatchPredict][google.cloud.automl.v1.PredictionService.BatchPredict]. name (str): diff --git a/google/cloud/automl_v1/services/prediction_service/transports/base.py b/google/cloud/automl_v1/services/prediction_service/transports/base.py index e11169b9..d6357ddd 100644 --- a/google/cloud/automl_v1/services/prediction_service/transports/base.py +++ b/google/cloud/automl_v1/services/prediction_service/transports/base.py @@ -117,7 +117,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/automl_v1/services/prediction_service/transports/grpc.py b/google/cloud/automl_v1/services/prediction_service/transports/grpc.py index 526028ad..ff0e33ac 100644 --- a/google/cloud/automl_v1/services/prediction_service/transports/grpc.py +++ b/google/cloud/automl_v1/services/prediction_service/transports/grpc.py @@ -85,16 +85,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1/services/prediction_service/transports/grpc_asyncio.py b/google/cloud/automl_v1/services/prediction_service/transports/grpc_asyncio.py index 629a6c02..1866b9b3 100644 --- a/google/cloud/automl_v1/services/prediction_service/transports/grpc_asyncio.py +++ b/google/cloud/automl_v1/services/prediction_service/transports/grpc_asyncio.py @@ -132,16 +132,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1beta1/services/auto_ml/client.py b/google/cloud/automl_v1beta1/services/auto_ml/client.py index c31da67d..5984253b 100644 --- a/google/cloud/automl_v1beta1/services/auto_ml/client.py +++ b/google/cloud/automl_v1beta1/services/auto_ml/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources from google.api_core import client_options as client_options_lib # type: ignore @@ -491,7 +491,7 @@ def __init__( def create_dataset( self, - request: service.CreateDatasetRequest = None, + request: Union[service.CreateDatasetRequest, dict] = None, *, parent: str = None, dataset: gca_dataset.Dataset = None, @@ -502,7 +502,7 @@ def create_dataset( r"""Creates a dataset. Args: - request (google.cloud.automl_v1beta1.types.CreateDatasetRequest): + request (Union[google.cloud.automl_v1beta1.types.CreateDatasetRequest, dict]): The request object. Request message for [AutoMl.CreateDataset][google.cloud.automl.v1beta1.AutoMl.CreateDataset]. parent (str): @@ -572,7 +572,7 @@ def create_dataset( def get_dataset( self, - request: service.GetDatasetRequest = None, + request: Union[service.GetDatasetRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -582,7 +582,7 @@ def get_dataset( r"""Gets a dataset. Args: - request (google.cloud.automl_v1beta1.types.GetDatasetRequest): + request (Union[google.cloud.automl_v1beta1.types.GetDatasetRequest, dict]): The request object. Request message for [AutoMl.GetDataset][google.cloud.automl.v1beta1.AutoMl.GetDataset]. name (str): @@ -645,7 +645,7 @@ def get_dataset( def list_datasets( self, - request: service.ListDatasetsRequest = None, + request: Union[service.ListDatasetsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -655,7 +655,7 @@ def list_datasets( r"""Lists datasets in a project. Args: - request (google.cloud.automl_v1beta1.types.ListDatasetsRequest): + request (Union[google.cloud.automl_v1beta1.types.ListDatasetsRequest, dict]): The request object. Request message for [AutoMl.ListDatasets][google.cloud.automl.v1beta1.AutoMl.ListDatasets]. parent (str): @@ -725,7 +725,7 @@ def list_datasets( def update_dataset( self, - request: service.UpdateDatasetRequest = None, + request: Union[service.UpdateDatasetRequest, dict] = None, *, dataset: gca_dataset.Dataset = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -735,7 +735,7 @@ def update_dataset( r"""Updates a dataset. Args: - request (google.cloud.automl_v1beta1.types.UpdateDatasetRequest): + request (Union[google.cloud.automl_v1beta1.types.UpdateDatasetRequest, dict]): The request object. Request message for [AutoMl.UpdateDataset][google.cloud.automl.v1beta1.AutoMl.UpdateDataset] dataset (google.cloud.automl_v1beta1.types.Dataset): @@ -800,7 +800,7 @@ def update_dataset( def delete_dataset( self, - request: service.DeleteDatasetRequest = None, + request: Union[service.DeleteDatasetRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -814,7 +814,7 @@ def delete_dataset( [metadata][google.longrunning.Operation.metadata] field. Args: - request (google.cloud.automl_v1beta1.types.DeleteDatasetRequest): + request (Union[google.cloud.automl_v1beta1.types.DeleteDatasetRequest, dict]): The request object. Request message for [AutoMl.DeleteDataset][google.cloud.automl.v1beta1.AutoMl.DeleteDataset]. name (str): @@ -896,7 +896,7 @@ def delete_dataset( def import_data( self, - request: service.ImportDataRequest = None, + request: Union[service.ImportDataRequest, dict] = None, *, name: str = None, input_config: io.InputConfig = None, @@ -916,7 +916,7 @@ def import_data( field when it completes. Args: - request (google.cloud.automl_v1beta1.types.ImportDataRequest): + request (Union[google.cloud.automl_v1beta1.types.ImportDataRequest, dict]): The request object. Request message for [AutoMl.ImportData][google.cloud.automl.v1beta1.AutoMl.ImportData]. name (str): @@ -1009,7 +1009,7 @@ def import_data( def export_data( self, - request: service.ExportDataRequest = None, + request: Union[service.ExportDataRequest, dict] = None, *, name: str = None, output_config: io.OutputConfig = None, @@ -1023,7 +1023,7 @@ def export_data( completes. Args: - request (google.cloud.automl_v1beta1.types.ExportDataRequest): + request (Union[google.cloud.automl_v1beta1.types.ExportDataRequest, dict]): The request object. Request message for [AutoMl.ExportData][google.cloud.automl.v1beta1.AutoMl.ExportData]. name (str): @@ -1114,7 +1114,7 @@ def export_data( def get_annotation_spec( self, - request: service.GetAnnotationSpecRequest = None, + request: Union[service.GetAnnotationSpecRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1124,7 +1124,7 @@ def get_annotation_spec( r"""Gets an annotation spec. Args: - request (google.cloud.automl_v1beta1.types.GetAnnotationSpecRequest): + request (Union[google.cloud.automl_v1beta1.types.GetAnnotationSpecRequest, dict]): The request object. Request message for [AutoMl.GetAnnotationSpec][google.cloud.automl.v1beta1.AutoMl.GetAnnotationSpec]. name (str): @@ -1183,7 +1183,7 @@ def get_annotation_spec( def get_table_spec( self, - request: service.GetTableSpecRequest = None, + request: Union[service.GetTableSpecRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1193,7 +1193,7 @@ def get_table_spec( r"""Gets a table spec. Args: - request (google.cloud.automl_v1beta1.types.GetTableSpecRequest): + request (Union[google.cloud.automl_v1beta1.types.GetTableSpecRequest, dict]): The request object. Request message for [AutoMl.GetTableSpec][google.cloud.automl.v1beta1.AutoMl.GetTableSpec]. name (str): @@ -1262,7 +1262,7 @@ def get_table_spec( def list_table_specs( self, - request: service.ListTableSpecsRequest = None, + request: Union[service.ListTableSpecsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1272,7 +1272,7 @@ def list_table_specs( r"""Lists table specs in a dataset. Args: - request (google.cloud.automl_v1beta1.types.ListTableSpecsRequest): + request (Union[google.cloud.automl_v1beta1.types.ListTableSpecsRequest, dict]): The request object. Request message for [AutoMl.ListTableSpecs][google.cloud.automl.v1beta1.AutoMl.ListTableSpecs]. parent (str): @@ -1342,7 +1342,7 @@ def list_table_specs( def update_table_spec( self, - request: service.UpdateTableSpecRequest = None, + request: Union[service.UpdateTableSpecRequest, dict] = None, *, table_spec: gca_table_spec.TableSpec = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1352,7 +1352,7 @@ def update_table_spec( r"""Updates a table spec. Args: - request (google.cloud.automl_v1beta1.types.UpdateTableSpecRequest): + request (Union[google.cloud.automl_v1beta1.types.UpdateTableSpecRequest, dict]): The request object. Request message for [AutoMl.UpdateTableSpec][google.cloud.automl.v1beta1.AutoMl.UpdateTableSpec] table_spec (google.cloud.automl_v1beta1.types.TableSpec): @@ -1423,7 +1423,7 @@ def update_table_spec( def get_column_spec( self, - request: service.GetColumnSpecRequest = None, + request: Union[service.GetColumnSpecRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1433,7 +1433,7 @@ def get_column_spec( r"""Gets a column spec. Args: - request (google.cloud.automl_v1beta1.types.GetColumnSpecRequest): + request (Union[google.cloud.automl_v1beta1.types.GetColumnSpecRequest, dict]): The request object. Request message for [AutoMl.GetColumnSpec][google.cloud.automl.v1beta1.AutoMl.GetColumnSpec]. name (str): @@ -1494,7 +1494,7 @@ def get_column_spec( def list_column_specs( self, - request: service.ListColumnSpecsRequest = None, + request: Union[service.ListColumnSpecsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1504,7 +1504,7 @@ def list_column_specs( r"""Lists column specs in a table spec. Args: - request (google.cloud.automl_v1beta1.types.ListColumnSpecsRequest): + request (Union[google.cloud.automl_v1beta1.types.ListColumnSpecsRequest, dict]): The request object. Request message for [AutoMl.ListColumnSpecs][google.cloud.automl.v1beta1.AutoMl.ListColumnSpecs]. parent (str): @@ -1574,7 +1574,7 @@ def list_column_specs( def update_column_spec( self, - request: service.UpdateColumnSpecRequest = None, + request: Union[service.UpdateColumnSpecRequest, dict] = None, *, column_spec: gca_column_spec.ColumnSpec = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1584,7 +1584,7 @@ def update_column_spec( r"""Updates a column spec. Args: - request (google.cloud.automl_v1beta1.types.UpdateColumnSpecRequest): + request (Union[google.cloud.automl_v1beta1.types.UpdateColumnSpecRequest, dict]): The request object. Request message for [AutoMl.UpdateColumnSpec][google.cloud.automl.v1beta1.AutoMl.UpdateColumnSpec] column_spec (google.cloud.automl_v1beta1.types.ColumnSpec): @@ -1647,7 +1647,7 @@ def update_column_spec( def create_model( self, - request: service.CreateModelRequest = None, + request: Union[service.CreateModelRequest, dict] = None, *, parent: str = None, model: gca_model.Model = None, @@ -1662,7 +1662,7 @@ def create_model( each annotation spec. Args: - request (google.cloud.automl_v1beta1.types.CreateModelRequest): + request (Union[google.cloud.automl_v1beta1.types.CreateModelRequest, dict]): The request object. Request message for [AutoMl.CreateModel][google.cloud.automl.v1beta1.AutoMl.CreateModel]. parent (str): @@ -1742,7 +1742,7 @@ def create_model( def get_model( self, - request: service.GetModelRequest = None, + request: Union[service.GetModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1752,7 +1752,7 @@ def get_model( r"""Gets a model. Args: - request (google.cloud.automl_v1beta1.types.GetModelRequest): + request (Union[google.cloud.automl_v1beta1.types.GetModelRequest, dict]): The request object. Request message for [AutoMl.GetModel][google.cloud.automl.v1beta1.AutoMl.GetModel]. name (str): @@ -1811,7 +1811,7 @@ def get_model( def list_models( self, - request: service.ListModelsRequest = None, + request: Union[service.ListModelsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1821,7 +1821,7 @@ def list_models( r"""Lists models. Args: - request (google.cloud.automl_v1beta1.types.ListModelsRequest): + request (Union[google.cloud.automl_v1beta1.types.ListModelsRequest, dict]): The request object. Request message for [AutoMl.ListModels][google.cloud.automl.v1beta1.AutoMl.ListModels]. parent (str): @@ -1891,7 +1891,7 @@ def list_models( def delete_model( self, - request: service.DeleteModelRequest = None, + request: Union[service.DeleteModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -1904,7 +1904,7 @@ def delete_model( [metadata][google.longrunning.Operation.metadata] field. Args: - request (google.cloud.automl_v1beta1.types.DeleteModelRequest): + request (Union[google.cloud.automl_v1beta1.types.DeleteModelRequest, dict]): The request object. Request message for [AutoMl.DeleteModel][google.cloud.automl.v1beta1.AutoMl.DeleteModel]. name (str): @@ -1986,7 +1986,7 @@ def delete_model( def deploy_model( self, - request: service.DeployModelRequest = None, + request: Union[service.DeployModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -2010,7 +2010,7 @@ def deploy_model( completes. Args: - request (google.cloud.automl_v1beta1.types.DeployModelRequest): + request (Union[google.cloud.automl_v1beta1.types.DeployModelRequest, dict]): The request object. Request message for [AutoMl.DeployModel][google.cloud.automl.v1beta1.AutoMl.DeployModel]. name (str): @@ -2092,7 +2092,7 @@ def deploy_model( def undeploy_model( self, - request: service.UndeployModelRequest = None, + request: Union[service.UndeployModelRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -2110,7 +2110,7 @@ def undeploy_model( completes. Args: - request (google.cloud.automl_v1beta1.types.UndeployModelRequest): + request (Union[google.cloud.automl_v1beta1.types.UndeployModelRequest, dict]): The request object. Request message for [AutoMl.UndeployModel][google.cloud.automl.v1beta1.AutoMl.UndeployModel]. name (str): @@ -2192,7 +2192,7 @@ def undeploy_model( def export_model( self, - request: service.ExportModelRequest = None, + request: Union[service.ExportModelRequest, dict] = None, *, name: str = None, output_config: io.ModelExportOutputConfig = None, @@ -2211,7 +2211,7 @@ def export_model( completes. Args: - request (google.cloud.automl_v1beta1.types.ExportModelRequest): + request (Union[google.cloud.automl_v1beta1.types.ExportModelRequest, dict]): The request object. Request message for [AutoMl.ExportModel][google.cloud.automl.v1beta1.AutoMl.ExportModel]. Models need to be enabled for exporting, otherwise an @@ -2304,7 +2304,7 @@ def export_model( def export_evaluated_examples( self, - request: service.ExportEvaluatedExamplesRequest = None, + request: Union[service.ExportEvaluatedExamplesRequest, dict] = None, *, name: str = None, output_config: io.ExportEvaluatedExamplesOutputConfig = None, @@ -2329,7 +2329,7 @@ def export_evaluated_examples( completes. Args: - request (google.cloud.automl_v1beta1.types.ExportEvaluatedExamplesRequest): + request (Union[google.cloud.automl_v1beta1.types.ExportEvaluatedExamplesRequest, dict]): The request object. Request message for [AutoMl.ExportEvaluatedExamples][google.cloud.automl.v1beta1.AutoMl.ExportEvaluatedExamples]. name (str): @@ -2423,7 +2423,7 @@ def export_evaluated_examples( def get_model_evaluation( self, - request: service.GetModelEvaluationRequest = None, + request: Union[service.GetModelEvaluationRequest, dict] = None, *, name: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -2433,7 +2433,7 @@ def get_model_evaluation( r"""Gets a model evaluation. Args: - request (google.cloud.automl_v1beta1.types.GetModelEvaluationRequest): + request (Union[google.cloud.automl_v1beta1.types.GetModelEvaluationRequest, dict]): The request object. Request message for [AutoMl.GetModelEvaluation][google.cloud.automl.v1beta1.AutoMl.GetModelEvaluation]. name (str): @@ -2492,7 +2492,7 @@ def get_model_evaluation( def list_model_evaluations( self, - request: service.ListModelEvaluationsRequest = None, + request: Union[service.ListModelEvaluationsRequest, dict] = None, *, parent: str = None, retry: retries.Retry = gapic_v1.method.DEFAULT, @@ -2502,7 +2502,7 @@ def list_model_evaluations( r"""Lists model evaluations. Args: - request (google.cloud.automl_v1beta1.types.ListModelEvaluationsRequest): + request (Union[google.cloud.automl_v1beta1.types.ListModelEvaluationsRequest, dict]): The request object. Request message for [AutoMl.ListModelEvaluations][google.cloud.automl.v1beta1.AutoMl.ListModelEvaluations]. parent (str): diff --git a/google/cloud/automl_v1beta1/services/auto_ml/transports/base.py b/google/cloud/automl_v1beta1/services/auto_ml/transports/base.py index 65d1e0c9..42506d98 100644 --- a/google/cloud/automl_v1beta1/services/auto_ml/transports/base.py +++ b/google/cloud/automl_v1beta1/services/auto_ml/transports/base.py @@ -126,7 +126,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc.py b/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc.py index fb5a3127..4e3ee6e4 100644 --- a/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc.py +++ b/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc.py @@ -105,16 +105,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc_asyncio.py b/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc_asyncio.py index 0ad45a69..8164fdf3 100644 --- a/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc_asyncio.py +++ b/google/cloud/automl_v1beta1/services/auto_ml/transports/grpc_asyncio.py @@ -152,16 +152,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1beta1/services/prediction_service/client.py b/google/cloud/automl_v1beta1/services/prediction_service/client.py index 1d9ca992..855ec386 100644 --- a/google/cloud/automl_v1beta1/services/prediction_service/client.py +++ b/google/cloud/automl_v1beta1/services/prediction_service/client.py @@ -17,7 +17,7 @@ from distutils import util import os import re -from typing import Callable, Dict, Optional, Sequence, Tuple, Type, Union +from typing import Dict, Optional, Sequence, Tuple, Type, Union import pkg_resources from google.api_core import client_options as client_options_lib # type: ignore @@ -363,7 +363,7 @@ def __init__( def predict( self, - request: prediction_service.PredictRequest = None, + request: Union[prediction_service.PredictRequest, dict] = None, *, name: str = None, payload: data_items.ExamplePayload = None, @@ -395,7 +395,7 @@ def predict( UTF-8 encoded. Args: - request (google.cloud.automl_v1beta1.types.PredictRequest): + request (Union[google.cloud.automl_v1beta1.types.PredictRequest, dict]): The request object. Request message for [PredictionService.Predict][google.cloud.automl.v1beta1.PredictionService.Predict]. name (str): @@ -496,7 +496,7 @@ def predict( def batch_predict( self, - request: prediction_service.BatchPredictRequest = None, + request: Union[prediction_service.BatchPredictRequest, dict] = None, *, name: str = None, input_config: io.BatchPredictInputConfig = None, @@ -525,7 +525,7 @@ def batch_predict( - Tables Args: - request (google.cloud.automl_v1beta1.types.BatchPredictRequest): + request (Union[google.cloud.automl_v1beta1.types.BatchPredictRequest, dict]): The request object. Request message for [PredictionService.BatchPredict][google.cloud.automl.v1beta1.PredictionService.BatchPredict]. name (str): diff --git a/google/cloud/automl_v1beta1/services/prediction_service/transports/base.py b/google/cloud/automl_v1beta1/services/prediction_service/transports/base.py index 2ea40f0a..af685b4e 100644 --- a/google/cloud/automl_v1beta1/services/prediction_service/transports/base.py +++ b/google/cloud/automl_v1beta1/services/prediction_service/transports/base.py @@ -117,7 +117,7 @@ def __init__( **scopes_kwargs, quota_project_id=quota_project_id ) - # If the credentials is service account credentials, then always try to use self signed JWT. + # If the credentials are service account credentials, then always try to use self signed JWT. if ( always_use_jwt_access and isinstance(credentials, service_account.Credentials) diff --git a/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc.py b/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc.py index 46f0611f..59f2f81f 100644 --- a/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc.py +++ b/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc.py @@ -85,16 +85,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc_asyncio.py b/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc_asyncio.py index 24ac8d4c..2ecadd9a 100644 --- a/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc_asyncio.py +++ b/google/cloud/automl_v1beta1/services/prediction_service/transports/grpc_asyncio.py @@ -132,16 +132,16 @@ def __init__( api_mtls_endpoint (Optional[str]): Deprecated. The mutual TLS endpoint. If provided, it overrides the ``host`` argument and tries to create a mutual TLS channel with client SSL credentials from - ``client_cert_source`` or applicatin default SSL credentials. + ``client_cert_source`` or application default SSL credentials. client_cert_source (Optional[Callable[[], Tuple[bytes, bytes]]]): Deprecated. A callback to provide client SSL certificate bytes and private key bytes, both in PEM format. It is ignored if ``api_mtls_endpoint`` is None. ssl_channel_credentials (grpc.ChannelCredentials): SSL credentials - for grpc channel. It is ignored if ``channel`` is provided. + for the grpc channel. It is ignored if ``channel`` is provided. client_cert_source_for_mtls (Optional[Callable[[], Tuple[bytes, bytes]]]): A callback to provide client certificate bytes and private key bytes, - both in PEM format. It is used to configure mutual TLS channel. It is + both in PEM format. It is used to configure a mutual TLS channel. It is ignored if ``channel`` or ``ssl_channel_credentials`` is provided. quota_project_id (Optional[str]): An optional project to use for billing and quota. diff --git a/scripts/fixup_automl_v1_keywords.py b/scripts/fixup_automl_v1_keywords.py index be4c58a4..ce01fcc8 100644 --- a/scripts/fixup_automl_v1_keywords.py +++ b/scripts/fixup_automl_v1_keywords.py @@ -39,26 +39,26 @@ def partition( class automlCallTransformer(cst.CSTTransformer): CTRL_PARAMS: Tuple[str] = ('retry', 'timeout', 'metadata') METHOD_TO_PARAMS: Dict[str, Tuple[str]] = { - 'batch_predict': ('name', 'input_config', 'output_config', 'params', ), - 'create_dataset': ('parent', 'dataset', ), - 'create_model': ('parent', 'model', ), - 'delete_dataset': ('name', ), - 'delete_model': ('name', ), - 'deploy_model': ('name', 'image_object_detection_model_deployment_metadata', 'image_classification_model_deployment_metadata', ), - 'export_data': ('name', 'output_config', ), - 'export_model': ('name', 'output_config', ), - 'get_annotation_spec': ('name', ), - 'get_dataset': ('name', ), - 'get_model': ('name', ), - 'get_model_evaluation': ('name', ), - 'import_data': ('name', 'input_config', ), - 'list_datasets': ('parent', 'filter', 'page_size', 'page_token', ), - 'list_model_evaluations': ('parent', 'filter', 'page_size', 'page_token', ), - 'list_models': ('parent', 'filter', 'page_size', 'page_token', ), - 'predict': ('name', 'payload', 'params', ), - 'undeploy_model': ('name', ), - 'update_dataset': ('dataset', 'update_mask', ), - 'update_model': ('model', 'update_mask', ), + 'batch_predict': ('name', 'input_config', 'output_config', 'params', ), + 'create_dataset': ('parent', 'dataset', ), + 'create_model': ('parent', 'model', ), + 'delete_dataset': ('name', ), + 'delete_model': ('name', ), + 'deploy_model': ('name', 'image_object_detection_model_deployment_metadata', 'image_classification_model_deployment_metadata', ), + 'export_data': ('name', 'output_config', ), + 'export_model': ('name', 'output_config', ), + 'get_annotation_spec': ('name', ), + 'get_dataset': ('name', ), + 'get_model': ('name', ), + 'get_model_evaluation': ('name', ), + 'import_data': ('name', 'input_config', ), + 'list_datasets': ('parent', 'filter', 'page_size', 'page_token', ), + 'list_model_evaluations': ('parent', 'filter', 'page_size', 'page_token', ), + 'list_models': ('parent', 'filter', 'page_size', 'page_token', ), + 'predict': ('name', 'payload', 'params', ), + 'undeploy_model': ('name', ), + 'update_dataset': ('dataset', 'update_mask', ), + 'update_model': ('model', 'update_mask', ), } def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: @@ -77,7 +77,7 @@ def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: return updated kwargs, ctrl_kwargs = partition( - lambda a: not a.keyword.value in self.CTRL_PARAMS, + lambda a: a.keyword.value not in self.CTRL_PARAMS, kwargs ) diff --git a/scripts/fixup_automl_v1beta1_keywords.py b/scripts/fixup_automl_v1beta1_keywords.py index 68569e00..e8e32cab 100644 --- a/scripts/fixup_automl_v1beta1_keywords.py +++ b/scripts/fixup_automl_v1beta1_keywords.py @@ -39,32 +39,32 @@ def partition( class automlCallTransformer(cst.CSTTransformer): CTRL_PARAMS: Tuple[str] = ('retry', 'timeout', 'metadata') METHOD_TO_PARAMS: Dict[str, Tuple[str]] = { - 'batch_predict': ('name', 'input_config', 'output_config', 'params', ), - 'create_dataset': ('parent', 'dataset', ), - 'create_model': ('parent', 'model', ), - 'delete_dataset': ('name', ), - 'delete_model': ('name', ), - 'deploy_model': ('name', 'image_object_detection_model_deployment_metadata', 'image_classification_model_deployment_metadata', ), - 'export_data': ('name', 'output_config', ), - 'export_evaluated_examples': ('name', 'output_config', ), - 'export_model': ('name', 'output_config', ), - 'get_annotation_spec': ('name', ), - 'get_column_spec': ('name', 'field_mask', ), - 'get_dataset': ('name', ), - 'get_model': ('name', ), - 'get_model_evaluation': ('name', ), - 'get_table_spec': ('name', 'field_mask', ), - 'import_data': ('name', 'input_config', ), - 'list_column_specs': ('parent', 'field_mask', 'filter', 'page_size', 'page_token', ), - 'list_datasets': ('parent', 'filter', 'page_size', 'page_token', ), - 'list_model_evaluations': ('parent', 'filter', 'page_size', 'page_token', ), - 'list_models': ('parent', 'filter', 'page_size', 'page_token', ), - 'list_table_specs': ('parent', 'field_mask', 'filter', 'page_size', 'page_token', ), - 'predict': ('name', 'payload', 'params', ), - 'undeploy_model': ('name', ), - 'update_column_spec': ('column_spec', 'update_mask', ), - 'update_dataset': ('dataset', 'update_mask', ), - 'update_table_spec': ('table_spec', 'update_mask', ), + 'batch_predict': ('name', 'input_config', 'output_config', 'params', ), + 'create_dataset': ('parent', 'dataset', ), + 'create_model': ('parent', 'model', ), + 'delete_dataset': ('name', ), + 'delete_model': ('name', ), + 'deploy_model': ('name', 'image_object_detection_model_deployment_metadata', 'image_classification_model_deployment_metadata', ), + 'export_data': ('name', 'output_config', ), + 'export_evaluated_examples': ('name', 'output_config', ), + 'export_model': ('name', 'output_config', ), + 'get_annotation_spec': ('name', ), + 'get_column_spec': ('name', 'field_mask', ), + 'get_dataset': ('name', ), + 'get_model': ('name', ), + 'get_model_evaluation': ('name', ), + 'get_table_spec': ('name', 'field_mask', ), + 'import_data': ('name', 'input_config', ), + 'list_column_specs': ('parent', 'field_mask', 'filter', 'page_size', 'page_token', ), + 'list_datasets': ('parent', 'filter', 'page_size', 'page_token', ), + 'list_model_evaluations': ('parent', 'filter', 'page_size', 'page_token', ), + 'list_models': ('parent', 'filter', 'page_size', 'page_token', ), + 'list_table_specs': ('parent', 'field_mask', 'filter', 'page_size', 'page_token', ), + 'predict': ('name', 'payload', 'params', ), + 'undeploy_model': ('name', ), + 'update_column_spec': ('column_spec', 'update_mask', ), + 'update_dataset': ('dataset', 'update_mask', ), + 'update_table_spec': ('table_spec', 'update_mask', ), } def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: @@ -83,7 +83,7 @@ def leave_Call(self, original: cst.Call, updated: cst.Call) -> cst.CSTNode: return updated kwargs, ctrl_kwargs = partition( - lambda a: not a.keyword.value in self.CTRL_PARAMS, + lambda a: a.keyword.value not in self.CTRL_PARAMS, kwargs )