🦉 Updates from OwlBot post-processor

See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md

google/cloud/aiplatform_v1/services/dataset_service/async_client.py

@@ -2334,7 +2334,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "DatasetServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/endpoint_service/async_client.py

@@ -2006,7 +2006,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "EndpointServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/featurestore_online_serving_service/async_client.py

@@ -1289,7 +1289,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "FeaturestoreOnlineServingServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/featurestore_service/async_client.py

@@ -3768,7 +3768,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "FeaturestoreServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/index_endpoint_service/async_client.py

@@ -1931,7 +1931,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "IndexEndpointServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/index_service/async_client.py

@@ -1684,7 +1684,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "IndexServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/job_service/async_client.py

@@ -5045,7 +5045,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "JobServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/match_service/async_client.py

@@ -1069,7 +1069,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MatchServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/metadata_service/async_client.py

@@ -4722,7 +4722,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MetadataServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/migration_service/async_client.py

@@ -1186,7 +1186,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MigrationServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/migration_service/client.py

@@ -208,40 +208,40 @@ def parse_annotated_dataset_path(path: str) -> Dict[str, str]:
     @staticmethod
     def dataset_path(
         project: str,
+        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/datasets/{dataset}".format(
+        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
             project=project,
+            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
+        m = re.match(
+            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
+            path,
+        )
         return m.groupdict() if m else {}
 
     @staticmethod
     def dataset_path(
         project: str,
-        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
+        return "projects/{project}/datasets/{dataset}".format(
             project=project,
-            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(
-            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
-            path,
-        )
+        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
         return m.groupdict() if m else {}
 
     @staticmethod

google/cloud/aiplatform_v1/services/model_garden_service/async_client.py

@@ -1009,7 +1009,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "ModelGardenServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/model_service/async_client.py

@@ -3125,7 +3125,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "ModelServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/pipeline_service/async_client.py

@@ -2102,7 +2102,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "PipelineServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/prediction_service/async_client.py

@@ -1394,7 +1394,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "PredictionServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/specialist_pool_service/async_client.py

@@ -1563,7 +1563,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "SpecialistPoolServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/tensorboard_service/async_client.py

@@ -4487,7 +4487,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "TensorboardServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1/services/vizier_service/async_client.py

@@ -2433,7 +2433,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "VizierServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/dataset_service/async_client.py

@@ -2334,7 +2334,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "DatasetServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/deployment_resource_pool_service/async_client.py

@@ -1579,7 +1579,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "DeploymentResourcePoolServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/endpoint_service/async_client.py

@@ -2012,7 +2012,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "EndpointServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/featurestore_online_serving_service/async_client.py

@@ -1289,7 +1289,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "FeaturestoreOnlineServingServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/featurestore_service/async_client.py

@@ -3769,7 +3769,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "FeaturestoreServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/index_endpoint_service/async_client.py

@@ -1931,7 +1931,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "IndexEndpointServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/index_service/async_client.py

@@ -1684,7 +1684,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "IndexServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/job_service/async_client.py

@@ -5051,7 +5051,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "JobServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/match_service/async_client.py

@@ -1069,7 +1069,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MatchServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/metadata_service/async_client.py

@@ -4722,7 +4722,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MetadataServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/migration_service/async_client.py

@@ -1186,7 +1186,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "MigrationServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/migration_service/client.py

@@ -208,40 +208,40 @@ def parse_annotated_dataset_path(path: str) -> Dict[str, str]:
     @staticmethod
     def dataset_path(
         project: str,
+        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/datasets/{dataset}".format(
+        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
             project=project,
+            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
+        m = re.match(
+            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
+            path,
+        )
         return m.groupdict() if m else {}
 
     @staticmethod
     def dataset_path(
         project: str,
-        location: str,
         dataset: str,
     ) -> str:
         """Returns a fully-qualified dataset string."""
-        return "projects/{project}/locations/{location}/datasets/{dataset}".format(
+        return "projects/{project}/datasets/{dataset}".format(
             project=project,
-            location=location,
             dataset=dataset,
         )
 
     @staticmethod
     def parse_dataset_path(path: str) -> Dict[str, str]:
         """Parses a dataset path into its component segments."""
-        m = re.match(
-            r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
-            path,
-        )
+        m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
         return m.groupdict() if m else {}
 
     @staticmethod

google/cloud/aiplatform_v1beta1/services/model_garden_service/async_client.py

@@ -1009,7 +1009,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "ModelGardenServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/model_service/async_client.py

@@ -3126,7 +3126,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "ModelServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/pipeline_service/async_client.py

@@ -2104,7 +2104,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "PipelineServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/prediction_service/async_client.py

@@ -1394,7 +1394,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "PredictionServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):

google/cloud/aiplatform_v1beta1/services/schedule_service/async_client.py

@@ -1733,7 +1733,7 @@ async def list_locations(
         # Done; return the response.
         return response
 
-    async def __aenter__(self):
+    async def __aenter__(self) -> "ScheduleServiceAsyncClient":
         return self
 
     async def __aexit__(self, exc_type, exc, tb):