apache-airflow-providers-google 10.14.0rc1__py3-none-any.whl → 10.15.0rc1__py3-none-any.whl

airflow/providers/google/cloud/hooks/mlengine.py

@@ -97,7 +97,7 @@ class MLEngineHook(GoogleBaseHook):
 
     def get_conn(self) -> Resource:
         """
-        Retrieves the connection to MLEngine.
+        Retrieve the connection to MLEngine.
 
         :return: Google MLEngine services object.
         """
@@ -107,7 +107,7 @@ class MLEngineHook(GoogleBaseHook):
     @GoogleBaseHook.fallback_to_default_project_id
     def create_job(self, job: dict, project_id: str, use_existing_job_fn: Callable | None = None) -> dict:
         """
-        Launches a MLEngine job and wait for it to reach a terminal state.
+        Launch a MLEngine job and wait for it to reach a terminal state.
 
         :param project_id: The Google Cloud project id within which MLEngine
             job will be launched. If set to None or missing, the default project_id from the Google Cloud
@@ -169,7 +169,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ):
         """
-        Launches a MLEngine job and wait for it to reach a terminal state.
+        Launch a MLEngine job and wait for it to reach a terminal state.
 
         :param project_id: The Google Cloud project id within which MLEngine
             job will be launched. If set to None or missing, the default project_id from the Google Cloud
@@ -202,7 +202,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ) -> dict:
         """
-        Cancels a MLEngine job.
+        Cancel a MLEngine job.
 
         :param project_id: The Google Cloud project id within which MLEngine
             job will be cancelled. If set to None or missing, the default project_id from the Google Cloud
@@ -231,7 +231,7 @@ class MLEngineHook(GoogleBaseHook):
 
     def get_job(self, project_id: str, job_id: str) -> dict:
         """
-        Gets a MLEngine job based on the job id.
+        Get a MLEngine job based on the job id.
 
         :param project_id: The project in which the Job is located. If set to None or missing, the default
             project_id from the Google Cloud connection is used. (templated)
@@ -255,7 +255,7 @@ class MLEngineHook(GoogleBaseHook):
 
     def _wait_for_job_done(self, project_id: str, job_id: str, interval: int = 30):
         """
-        Waits for the Job to reach a terminal state.
+        Wait for the Job to reach a terminal state.
 
         This method will periodically check the job state until the job reach
         a terminal state.
@@ -284,7 +284,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ) -> dict:
         """
-        Creates the Version on Google Cloud ML Engine.
+        Create the Version on Google Cloud ML Engine.
 
         :param version_spec: A dictionary containing the information about the version. (templated)
         :param model_name: The name of the Google Cloud ML Engine model that the version belongs to.
@@ -320,7 +320,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ) -> dict:
         """
-        Sets a version to be the default. Blocks until finished.
+        Set a version to be the default. Blocks until finished.
 
         :param model_name: The name of the Google Cloud ML Engine model that the version belongs to.
             (templated)
@@ -351,7 +351,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ) -> list[dict]:
         """
-        Lists all available versions of a model. Blocks until finished.
+        List all available versions of a model. Blocks until finished.
 
         :param model_name: The name of the Google Cloud ML Engine model that the version
             belongs to. (templated)
@@ -387,7 +387,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ) -> dict:
         """
-        Deletes the given version of a model. Blocks until finished.
+        Delete the given version of a model. Blocks until finished.
 
         :param model_name: The name of the Google Cloud ML Engine model that the version
             belongs to. (templated)
@@ -467,7 +467,7 @@ class MLEngineHook(GoogleBaseHook):
         project_id: str,
     ) -> dict | None:
         """
-        Gets a Model. Blocks until finished.
+        Get a Model. Blocks until finished.
 
         :param model_name: The name of the model.
         :param project_id: The Google Cloud project name to which MLEngine model belongs. If set to None
@@ -581,7 +581,7 @@ class MLEngineAsyncHook(GoogleBaseAsyncHook):
         project_id: str | None = None,
     ) -> str | None:
         """
-        Polls for job status asynchronously using gcloud-aio.
+        Poll for job status asynchronously using gcloud-aio.
 
         Note that an OSError is raised when Job results are still pending.
         Exception means that Job finished with errors

airflow/providers/google/cloud/hooks/natural_language.py

@@ -74,7 +74,7 @@ class CloudNaturalLanguageHook(GoogleBaseHook):
 
     def get_conn(self) -> LanguageServiceClient:
         """
-        Retrieves connection to Cloud Natural Language service.
+        Retrieve connection to Cloud Natural Language service.
 
         :return: Cloud Natural Language service object
         """
@@ -92,7 +92,7 @@ class CloudNaturalLanguageHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> AnalyzeEntitiesResponse:
         """
-        Finds named entities in the text along with various properties.
+        Find named entities in the text along with various properties.
 
         Examples properties: entity types, salience, mentions for each entity, and others.
 

airflow/providers/google/cloud/hooks/os_login.py

@@ -78,7 +78,7 @@ class OSLoginHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> ImportSshPublicKeyResponse:
         """
-        Adds an SSH public key and returns the profile information.
+        Add an SSH public key and returns the profile information.
 
         Default POSIX account information is set when no username and UID exist as part of the login profile.
 

airflow/providers/google/cloud/hooks/pubsub.py

@@ -93,7 +93,7 @@ class PubSubHook(GoogleBaseHook):
 
     def get_conn(self) -> PublisherClient:
         """
-        Retrieves connection to Google Cloud Pub/Sub.
+        Retrieve connection to Google Cloud Pub/Sub.
 
         :return: Google Cloud Pub/Sub client object.
         """
@@ -118,7 +118,7 @@ class PubSubHook(GoogleBaseHook):
         project_id: str = PROVIDE_PROJECT_ID,
     ) -> None:
         """
-        Publishes messages to a Pub/Sub topic.
+        Publish messages to a Pub/Sub topic.
 
         :param topic: the Pub/Sub topic to which to publish; do not
             include the ``projects/{project}/topics/`` prefix.
@@ -191,7 +191,7 @@ class PubSubHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Creates a Pub/Sub topic, if it does not already exist.
+        Create a Pub/Sub topic, if it does not already exist.
 
         :param topic: the Pub/Sub topic name to create; do not
             include the ``projects/{project}/topics/`` prefix.
@@ -264,7 +264,7 @@ class PubSubHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Deletes a Pub/Sub topic if it exists.
+        Delete a Pub/Sub topic if it exists.
 
         :param topic: the Pub/Sub topic name to delete; do not
             include the ``projects/{project}/topics/`` prefix.
@@ -318,7 +318,7 @@ class PubSubHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> str:
         """
-        Creates a Pub/Sub subscription, if it does not already exist.
+        Create a Pub/Sub subscription, if it does not already exist.
 
         :param topic: the Pub/Sub topic name that the subscription will be bound
             to create; do not include the ``projects/{project}/subscriptions/`` prefix.
@@ -435,7 +435,7 @@ class PubSubHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Deletes a Pub/Sub subscription, if it exists.
+        Delete a Pub/Sub subscription, if it exists.
 
         :param subscription: the Pub/Sub subscription name to delete; do not
             include the ``projects/{project}/subscriptions/`` prefix.
@@ -483,7 +483,7 @@ class PubSubHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> list[ReceivedMessage]:
         """
-        Pulls up to ``max_messages`` messages from Pub/Sub subscription.
+        Pull up to ``max_messages`` messages from Pub/Sub subscription.
 
         :param subscription: the Pub/Sub subscription name to pull from; do not
             include the 'projects/{project}/topics/' prefix.
@@ -596,7 +596,7 @@ class PubSubAsyncHook(GoogleBaseAsyncHook):
 
     async def _get_subscriber_client(self) -> SubscriberAsyncClient:
         """
-        Returns async connection to the Google PubSub.
+        Return async connection to the Google PubSub.
 
         :return: Google Pub/Sub asynchronous client.
         """
@@ -671,7 +671,7 @@ class PubSubAsyncHook(GoogleBaseAsyncHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> list[ReceivedMessage]:
         """
-        Pulls up to ``max_messages`` messages from Pub/Sub subscription.
+        Pull up to ``max_messages`` messages from Pub/Sub subscription.
 
         :param subscription: the Pub/Sub subscription name to pull from; do not
             include the 'projects/{project}/topics/' prefix.

airflow/providers/google/cloud/hooks/secret_manager.py

@@ -63,7 +63,7 @@ class SecretsManagerHook(GoogleBaseHook):
 
     def get_conn(self) -> _SecretManagerClient:
         """
-        Retrieves the connection to Secret Manager.
+        Retrieve the connection to Secret Manager.
 
         :return: Secret Manager client.
         """

airflow/providers/google/cloud/hooks/spanner.py

@@ -76,7 +76,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
 
     def _get_client(self, project_id: str) -> Client:
         """
-        Provides a client for interacting with the Cloud Spanner API.
+        Provide a client for interacting with the Cloud Spanner API.
 
         :param project_id: The ID of the Google Cloud project.
         :return: Client
@@ -122,7 +122,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         project_id: str,
     ) -> Instance | None:
         """
-        Gets information about a particular instance.
+        Get information about a particular instance.
 
         :param project_id: Optional, The ID of the Google Cloud project that owns the Cloud Spanner
             database. If set to None or missing, the default project_id from the Google Cloud connection
@@ -145,7 +145,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         func: Callable[[Instance], Operation],
     ) -> None:
         """
-        Invokes a method on a given instance by applying a specified Callable.
+        Invoke a method on a given instance by applying a specified Callable.
 
         :param project_id: The ID of the Google Cloud project that owns the Cloud Spanner database.
         :param instance_id: The ID of the instance.
@@ -183,7 +183,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         project_id: str,
     ) -> None:
         """
-        Creates a new Cloud Spanner instance.
+        Create a new Cloud Spanner instance.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param configuration_name: The name of the instance configuration defining how the
@@ -213,7 +213,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         project_id: str,
     ) -> None:
         """
-        Updates an existing Cloud Spanner instance.
+        Update an existing Cloud Spanner instance.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param configuration_name: The name of the instance configuration defining how the
@@ -236,7 +236,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
     @GoogleBaseHook.fallback_to_default_project_id
     def delete_instance(self, instance_id: str, project_id: str) -> None:
         """
-        Deletes an existing Cloud Spanner instance.
+        Delete an existing Cloud Spanner instance.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param project_id: Optional, the ID of the Google Cloud project that owns the Cloud Spanner
@@ -260,7 +260,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         project_id: str,
     ) -> Database | None:
         """
-        Retrieves a database in Cloud Spanner; return None if the database does not exist in the instance.
+        Retrieve a database in Cloud Spanner; return None if the database does not exist in the instance.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param database_id: The ID of the database in Cloud Spanner.
@@ -287,7 +287,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         project_id: str,
     ) -> None:
         """
-        Creates a new database in Cloud Spanner.
+        Create a new database in Cloud Spanner.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param database_id: The ID of the database to create in Cloud Spanner.
@@ -321,7 +321,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         operation_id: str | None = None,
     ) -> None:
         """
-        Updates DDL of a database in Cloud Spanner.
+        Update DDL of a database in Cloud Spanner.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param database_id: The ID of the database in Cloud Spanner.
@@ -357,7 +357,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
     @GoogleBaseHook.fallback_to_default_project_id
     def delete_database(self, instance_id: str, database_id, project_id: str) -> bool:
         """
-        Drops a database in Cloud Spanner.
+        Drop a database in Cloud Spanner.
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param database_id: The ID of the database in Cloud Spanner.
@@ -392,7 +392,7 @@ class SpannerHook(GoogleBaseHook, DbApiHook):
         project_id: str,
     ) -> None:
         """
-        Executes an arbitrary DML query (INSERT, UPDATE, DELETE).
+        Execute an arbitrary DML query (INSERT, UPDATE, DELETE).
 
         :param instance_id: The ID of the Cloud Spanner instance.
         :param database_id: The ID of the database in Cloud Spanner.

airflow/providers/google/cloud/hooks/speech_to_text.py

@@ -65,7 +65,7 @@ class CloudSpeechToTextHook(GoogleBaseHook):
 
     def get_conn(self) -> SpeechClient:
         """
-        Retrieves connection to Cloud Speech.
+        Retrieve connection to Cloud Speech.
 
         :return: Google Cloud Speech client object.
         """

airflow/providers/google/cloud/hooks/stackdriver.py

@@ -80,7 +80,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Any:
         """
-        Fetches all the Alert Policies identified by the filter passed as filter parameter.
+        Fetch all the Alert Policies identified by the filter passed as filter parameter.
 
         The desired return type can be specified by the format parameter, the supported formats
         are "dict", "json" and None which returns python dictionary, stringified JSON and protobuf
@@ -161,7 +161,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Enables one or more disabled alerting policies identified by filter parameter.
+        Enable one or more disabled alerting policies identified by filter parameter.
 
         Inoperative in case the policy is already enabled.
 
@@ -229,7 +229,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-         Creates a new alert or updates an existing policy identified the name field in the alerts parameter.
+         Create a new alert or updates an existing policy identified the name field in the alerts parameter.
 
         :param project_id: The project in which alert needs to be created/updated.
         :param alerts: A JSON string or file that specifies all the alerts that needs
@@ -322,7 +322,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Deletes an alerting policy.
+        Delete an alerting policy.
 
         :param name: The alerting policy to delete. The format is:
                          ``projects/[PROJECT_ID]/alertPolicies/[ALERT_POLICY_ID]``.
@@ -354,7 +354,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Any:
         """
-        Fetches all the Notification Channels identified by the filter passed as filter parameter.
+        Fetch all the Notification Channels identified by the filter passed as filter parameter.
 
         The desired return type can be specified by the format parameter, the supported formats are
         "dict", "json" and None which returns python dictionary, stringified JSON and protobuf
@@ -437,7 +437,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Enables one or more disabled alerting policies identified by filter parameter.
+        Enable one or more disabled alerting policies identified by filter parameter.
 
         Inoperative in case the policy is already enabled.
 
@@ -567,7 +567,7 @@ class StackdriverHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Deletes a notification channel.
+        Delete a notification channel.
 
         :param name: The alerting policy to delete. The format is:
                          ``projects/[PROJECT_ID]/notificationChannels/[CHANNEL_ID]``.

airflow/providers/google/cloud/hooks/tasks.py

@@ -73,7 +73,7 @@ class CloudTasksHook(GoogleBaseHook):
 
     def get_conn(self) -> CloudTasksClient:
         """
-        Provides a client for interacting with the Google Cloud Tasks API.
+        Provide a client for interacting with the Google Cloud Tasks API.
 
         :return: Google Cloud Tasks API Client
         """
@@ -93,7 +93,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Queue:
         """
-        Creates a queue in Cloud Tasks.
+        Create a queue in Cloud Tasks.
 
         :param location: The location name in which the queue will be created.
         :param task_queue: The task queue to create.
@@ -141,7 +141,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Queue:
         """
-        Updates a queue in Cloud Tasks.
+        Update a queue in Cloud Tasks.
 
         :param task_queue: The task queue to update.
             This method creates the queue if it does not exist and updates the queue if
@@ -190,7 +190,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Queue:
         """
-        Gets a queue from Cloud Tasks.
+        Get a queue from Cloud Tasks.
 
         :param location: The location name in which the queue was created.
         :param queue_name: The queue's name.
@@ -225,7 +225,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> list[Queue]:
         """
-        Lists queues from Cloud Tasks.
+        List queues from Cloud Tasks.
 
         :param location: The location name in which the queues were created.
         :param project_id: (Optional) The ID of the Google Cloud project that owns the Cloud Tasks.
@@ -262,7 +262,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Deletes a queue from Cloud Tasks, even if it has tasks in it.
+        Delete a queue from Cloud Tasks, even if it has tasks in it.
 
         :param location: The location name in which the queue will be deleted.
         :param queue_name: The queue's name.
@@ -401,7 +401,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Task:
         """
-        Creates a task in Cloud Tasks.
+        Create a task in Cloud Tasks.
 
         :param location: The location name in which the task will be created.
         :param queue_name: The queue's name.
@@ -453,7 +453,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Task:
         """
-        Gets a task from Cloud Tasks.
+        Get a task from Cloud Tasks.
 
         :param location: The location name in which the task was created.
         :param queue_name: The queue's name.
@@ -492,7 +492,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> list[Task]:
         """
-        Lists the tasks in Cloud Tasks.
+        List the tasks in Cloud Tasks.
 
         :param location: The location name in which the tasks were created.
         :param queue_name: The queue's name.
@@ -531,7 +531,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> None:
         """
-        Deletes a task from Cloud Tasks.
+        Delete a task from Cloud Tasks.
 
         :param location: The location name in which the task will be deleted.
         :param queue_name: The queue's name.
@@ -568,7 +568,7 @@ class CloudTasksHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Task:
         """
-        Forces to run a task in Cloud Tasks.
+        Force run a task in Cloud Tasks.
 
         :param location: The location name in which the task was created.
         :param queue_name: The queue's name.

airflow/providers/google/cloud/hooks/text_to_speech.py

@@ -73,7 +73,7 @@ class CloudTextToSpeechHook(GoogleBaseHook):
 
     def get_conn(self) -> TextToSpeechClient:
         """
-        Retrieves connection to Cloud Text to Speech.
+        Retrieve connection to Cloud Text to Speech.
 
         :return: Google Cloud Text to Speech client object.
         """

airflow/providers/google/cloud/hooks/translate.py

@@ -53,7 +53,7 @@ class CloudTranslateHook(GoogleBaseHook):
 
     def get_conn(self) -> Client:
         """
-        Retrieves connection to Cloud Translate.
+        Retrieve connection to Cloud Translate.
 
         :return: Google Cloud Translate client object.
         """

airflow/providers/google/cloud/hooks/vertex_ai/auto_ml.py

@@ -74,7 +74,7 @@ class AutoMLHook(GoogleBaseHook):
         self,
         region: str | None = None,
     ) -> PipelineServiceClient:
-        """Returns PipelineServiceClient."""
+        """Return PipelineServiceClient."""
         if region and region != "global":
             client_options = ClientOptions(api_endpoint=f"{region}-aiplatform.googleapis.com:443")
         else:
@@ -88,7 +88,7 @@ class AutoMLHook(GoogleBaseHook):
         self,
         region: str | None = None,
     ) -> JobServiceClient:
-        """Returns JobServiceClient."""
+        """Return JobServiceClient."""
         if region and region != "global":
             client_options = ClientOptions(api_endpoint=f"{region}-aiplatform.googleapis.com:443")
         else:
@@ -113,7 +113,7 @@ class AutoMLHook(GoogleBaseHook):
         training_encryption_spec_key_name: str | None = None,
         model_encryption_spec_key_name: str | None = None,
     ) -> AutoMLTabularTrainingJob:
-        """Returns AutoMLTabularTrainingJob object."""
+        """Return AutoMLTabularTrainingJob object."""
         return AutoMLTabularTrainingJob(
             display_name=display_name,
             optimization_prediction_type=optimization_prediction_type,
@@ -142,7 +142,7 @@ class AutoMLHook(GoogleBaseHook):
         training_encryption_spec_key_name: str | None = None,
         model_encryption_spec_key_name: str | None = None,
     ) -> AutoMLForecastingTrainingJob:
-        """Returns AutoMLForecastingTrainingJob object."""
+        """Return AutoMLForecastingTrainingJob object."""
         return AutoMLForecastingTrainingJob(
             display_name=display_name,
             optimization_objective=optimization_objective,
@@ -169,7 +169,7 @@ class AutoMLHook(GoogleBaseHook):
         training_encryption_spec_key_name: str | None = None,
         model_encryption_spec_key_name: str | None = None,
     ) -> AutoMLImageTrainingJob:
-        """Returns AutoMLImageTrainingJob object."""
+        """Return AutoMLImageTrainingJob object."""
         return AutoMLImageTrainingJob(
             display_name=display_name,
             prediction_type=prediction_type,
@@ -196,7 +196,7 @@ class AutoMLHook(GoogleBaseHook):
         training_encryption_spec_key_name: str | None = None,
         model_encryption_spec_key_name: str | None = None,
     ) -> AutoMLTextTrainingJob:
-        """Returns AutoMLTextTrainingJob object."""
+        """Return AutoMLTextTrainingJob object."""
         return AutoMLTextTrainingJob(
             display_name=display_name,
             prediction_type=prediction_type,
@@ -221,7 +221,7 @@ class AutoMLHook(GoogleBaseHook):
         training_encryption_spec_key_name: str | None = None,
         model_encryption_spec_key_name: str | None = None,
     ) -> AutoMLVideoTrainingJob:
-        """Returns AutoMLVideoTrainingJob object."""
+        """Return AutoMLVideoTrainingJob object."""
         return AutoMLVideoTrainingJob(
             display_name=display_name,
             prediction_type=prediction_type,
@@ -236,16 +236,16 @@ class AutoMLHook(GoogleBaseHook):
 
     @staticmethod
     def extract_model_id(obj: dict) -> str:
-        """Returns unique id of the Model."""
+        """Return unique id of the Model."""
         return obj["name"].rpartition("/")[-1]
 
     @staticmethod
     def extract_training_id(resource_name: str) -> str:
-        """Returns unique id of the Training pipeline."""
+        """Return unique id of the Training pipeline."""
         return resource_name.rpartition("/")[-1]
 
     def wait_for_operation(self, operation: Operation, timeout: float | None = None):
-        """Waits for long-lasting operation to complete."""
+        """Wait for long-lasting operation to complete."""
         try:
             return operation.result(timeout=timeout)
         except Exception:
@@ -1301,7 +1301,7 @@ class AutoMLHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> Operation:
         """
-        Deletes a TrainingPipeline.
+        Delete a TrainingPipeline.
 
         :param project_id: Required. The ID of the Google Cloud project that the service belongs to.
         :param region: Required. The ID of the Google Cloud region that the service belongs to.
@@ -1334,7 +1334,7 @@ class AutoMLHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> TrainingPipeline:
         """
-        Gets a TrainingPipeline.
+        Get a TrainingPipeline.
 
         :param project_id: Required. The ID of the Google Cloud project that the service belongs to.
         :param region: Required. The ID of the Google Cloud region that the service belongs to.
@@ -1370,7 +1370,7 @@ class AutoMLHook(GoogleBaseHook):
         metadata: Sequence[tuple[str, str]] = (),
     ) -> ListTrainingPipelinesPager:
         """
-        Lists TrainingPipelines in a Location.
+        List TrainingPipelines in a Location.
 
         :param project_id: Required. The ID of the Google Cloud project that the service belongs to.
         :param region: Required. The ID of the Google Cloud region that the service belongs to.