🦉 Updates from OwlBot post-processor

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

Author: gcf-owl-bot[bot]

google/cloud/aiplatform_v1beta1/__init__.py

@@ -576,13 +576,19 @@
 from .types.pipeline_state import PipelineState
 from .types.prediction_service import CountTokensRequest
 from .types.prediction_service import CountTokensResponse
+from .types.prediction_service import DirectPredictRequest
+from .types.prediction_service import DirectPredictResponse
+from .types.prediction_service import DirectRawPredictRequest
+from .types.prediction_service import DirectRawPredictResponse
 from .types.prediction_service import ExplainRequest
 from .types.prediction_service import ExplainResponse
 from .types.prediction_service import PredictRequest
 from .types.prediction_service import PredictResponse
 from .types.prediction_service import RawPredictRequest
 from .types.prediction_service import StreamingPredictRequest
 from .types.prediction_service import StreamingPredictResponse
+from .types.prediction_service import StreamingRawPredictRequest
+from .types.prediction_service import StreamingRawPredictResponse
 from .types.publisher_model import PublisherModel
 from .types.saved_query import SavedQuery
 from .types.schedule import Schedule
@@ -911,6 +917,10 @@
     "DeploymentResourcePool",
     "DeploymentResourcePoolServiceClient",
     "DestinationFeatureSetting",
+    "DirectPredictRequest",
+    "DirectPredictResponse",
+    "DirectRawPredictRequest",
+    "DirectRawPredictResponse",
     "DiskSpec",
     "DoubleArray",
     "EncryptionSpec",
@@ -1272,6 +1282,8 @@
     "StratifiedSplit",
     "StreamingPredictRequest",
     "StreamingPredictResponse",
+    "StreamingRawPredictRequest",
+    "StreamingRawPredictResponse",
     "StreamingReadFeatureValuesRequest",
     "StringArray",
     "Study",

google/cloud/aiplatform_v1beta1/gapic_metadata.json

@@ -2311,6 +2311,16 @@
                 "count_tokens"
               ]
             },
+            "DirectPredict": {
+              "methods": [
+                "direct_predict"
+              ]
+            },
+            "DirectRawPredict": {
+              "methods": [
+                "direct_raw_predict"
+              ]
+            },
             "Explain": {
               "methods": [
                 "explain"
@@ -2330,6 +2340,16 @@
               "methods": [
                 "server_streaming_predict"
               ]
+            },
+            "StreamingPredict": {
+              "methods": [
+                "streaming_predict"
+              ]
+            },
+            "StreamingRawPredict": {
+              "methods": [
+                "streaming_raw_predict"
+              ]
             }
           }
         },
@@ -2341,6 +2361,16 @@
                 "count_tokens"
               ]
             },
+            "DirectPredict": {
+              "methods": [
+                "direct_predict"
+              ]
+            },
+            "DirectRawPredict": {
+              "methods": [
+                "direct_raw_predict"
+              ]
+            },
             "Explain": {
               "methods": [
                 "explain"
@@ -2360,6 +2390,16 @@
               "methods": [
                 "server_streaming_predict"
               ]
+            },
+            "StreamingPredict": {
+              "methods": [
+                "streaming_predict"
+              ]
+            },
+            "StreamingRawPredict": {
+              "methods": [
+                "streaming_raw_predict"
+              ]
             }
           }
         }

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

@@ -230,40 +230,40 @@ def parse_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