Copybara import of the project:

-- 222f6ea by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add UpdateExplanationDataset to aiplatform PiperOrigin-RevId: 541213716 Source-Link: googleapis/googleapis@4a858c4 Source-Link: googleapis/googleapis-gen@e4c3eab Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZTRjM2VhYjFmZGE0OWM1ZDk2MTliYjgyODcyYTBhN2U4MWY4MzUyOCJ9 -- 75bf15d by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- c8f2c7e by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: chore: Generate Ruby documentai wrapper from v1 instead of v1beta3 PiperOrigin-RevId: 542603310 Source-Link: googleapis/googleapis@45001d8 Source-Link: googleapis/googleapis-gen@ad6882b Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYWQ2ODgyYmI2NDhkMTE3MGE1NDE0ZTBlOTk5OWM4NGE0M2Y1MTVmMCJ9 -- df2e2d6 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- eb4f7ab by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: docs: tightened requirements on cloud storage subscription filename suffixes PiperOrigin-RevId: 543503563 Source-Link: googleapis/googleapis@212ecef Source-Link: googleapis/googleapis-gen@532d27e Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNTMyZDI3ZWVhNmZmMTI4ZmRjYmIzMzMyMTc2ZTg5NGU3NmRjYzY4NSJ9 -- 8b9d874 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 1ff05f7 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: fix: Add async context manager return types chore: Mock return_value should not populate oneof message fields chore: Support snippet generation for services that only support REST transport chore: Update gapic-generator-python to v1.11.0 PiperOrigin-RevId: 545430278 Source-Link: googleapis/googleapis@601b532 Source-Link: googleapis/googleapis-gen@b3f18d0 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiYjNmMThkMGY2NTYwYTg1NTAyMmZkMDU4ODY1ZTc2MjA0NzlkN2FmOSJ9 -- 79469ac by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 9c3dfe8 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: chore: Update gapic-generator-python to v1.11.1 PiperOrigin-RevId: 545769621 Source-Link: googleapis/googleapis@cc2f031 Source-Link: googleapis/googleapis-gen@1a46ae7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWE0NmFlNzY1MmVmOThhODZjMmJjMjMyYjZkYTA0MTJlOWYzMGJkYiJ9 -- f221309 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 043107e by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: added GitIntegrationSettings to the Agent PiperOrigin-RevId: 546355900 Source-Link: googleapis/googleapis@dc3d3a2 Source-Link: googleapis/googleapis-gen@2480a9f Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMjQ4MGE5ZjhiZmJjNzk4N2Y1ODNkNzBlMTk5NTg0YTQ2YjUyMTMzZiJ9 -- 5112a1a by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 57b17e5 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: chore: Update gapic-generator-python to v1.11.2 PiperOrigin-RevId: 546510849 Source-Link: googleapis/googleapis@736073a Source-Link: googleapis/googleapis-gen@deb64e8 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZGViNjRlOGVjMTlkMTQxZTMxMDg5ZmU5MzJiM2E5OTdhZDU0MWM0ZCJ9 -- 195ff4e by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 4c05c41 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: chore: Update gapic-generator-python to v1.11.3 PiperOrigin-RevId: 546899192 Source-Link: googleapis/googleapis@e6b1691 Source-Link: googleapis/googleapis-gen@0b3917c Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMGIzOTE3YzQyMWNiZGE3ZmNiNjcwOTJlMTZjMzNmM2VhNDZmNGJjNyJ9 -- e0cd533 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 6e490f4 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: Add routing information in Cloud Build GRPC clients PiperOrigin-RevId: 547283196 Source-Link: googleapis/googleapis@64d54ff Source-Link: googleapis/googleapis-gen@1de1822 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiMWRlMTgyMjExMDliNGEwMjVjMGIyZThjMjUzMjRjYjAxYjZkZjYyNiJ9 -- 57a75ff by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 1f1e8a4 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add data_item_count to Dataset feat: add DeleteSavedQuery to DatasetService feat: add JOB_STATE_PARTIALLY_SUCCEEDED to JobState feat: add pipeline_job to Model feat: add reserved_ip_ranges to PipelineJob PiperOrigin-RevId: 547343567 Source-Link: googleapis/googleapis@72b20cc Source-Link: googleapis/googleapis-gen@4d84e76 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNGQ4NGU3NmRmNmY1YTM2MWVjYTM1NzNhMGY2OTIyMWI0MjFlNzM1OSJ9 -- 61d7cf5 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 330ca6b by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add data_item_count to Dataset feat: add DeleteSavedQuery to DatasetService feat: add ExportEndpointOperationMetadata and ExportEndpointResponse to Endpoint feat: add JOB_STATE_PARTIALLY_SUCCEEDED to JobState feat: add pipeline_job to Model feat: add persistent_resource.proto and persistent_resource_service.proto feat: add reserved_ip_ranges to PipelineJob feat: add ReadTensorboardSize to TensorboardService PiperOrigin-RevId: 547519544 Source-Link: googleapis/googleapis@4f40e53 Source-Link: googleapis/googleapis-gen@cf937c7 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiY2Y5MzdjN2VjYmU4Y2VlYWEwMzFmMTM4OTkzNDE0OWJkYWRmMDRiZSJ9 -- acb959d by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 216a83e by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: chore: Update gapic-generator-python to v1.11.4 PiperOrigin-RevId: 547897126 Source-Link: googleapis/googleapis@c09c75e Source-Link: googleapis/googleapis-gen@45e0ec4 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNDVlMGVjNDM0MzUxN2NkMGFhNjZiNWNhNjQyMzJhMTgwMmMyZjk0NSJ9 -- 09bc5c6 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- ec20ac3 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: docs: small documentation updates chore: add enums for additional platform logs PiperOrigin-RevId: 549453285 Source-Link: googleapis/googleapis@b9fb22c Source-Link: googleapis/googleapis-gen@ea3ae98 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiZWEzYWU5ODAzMGI1ZTA0ZTcyODhhMmE5YmQzZWNhODE2NjM0OTFiZSJ9 -- 8ff3cd7 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 8e541cf by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: ScheduleService (schedule_service.proto) creates and manages Schedule resources to launch scheduled pipelines runs feat: Schedule (schedule.proto) periodically schedules runs to make API calls PiperOrigin-RevId: 550058610 Source-Link: googleapis/googleapis@5b24e63 Source-Link: googleapis/googleapis-gen@4c5f6f9 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNGM1ZjZmOWM0MjkyYzE0Mjg3MjE3YjU2ZjJkY2IwNGNkNzcyNzgzMiJ9 -- 8173e33 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- 6a3491c by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: feat: add policy_name to PlacementPolicy message within a node pool feat: add support for HostMaintenancePolicy feat: add support for AdditionalPodNetworkConfig and AdditionalNodeNetworkConfig feat: add enable_multi_networking to NetworkConfig PiperOrigin-RevId: 550168784 Source-Link: googleapis/googleapis@b53aeac Source-Link: googleapis/googleapis-gen@6e0edf4 Copy-Tag: eyJwIjoiLmdpdGh1Yi8uT3dsQm90LnlhbWwiLCJoIjoiNmUwZWRmNDhjOWQ5OTFhMmQwMmFhOTRlNzIzYmJkMDdlNDVlMTRkNCJ9 -- 72fbdb9 by Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>: 🦉 Updates from OwlBot post-processor See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md -- ef2bc05 by nayaknishant <[email protected]>: reverting Owlbot flake8 changes -- ddfe450 by nayaknishant <[email protected]>: removing merge HEAD comment COPYBARA_INTEGRATE_REVIEW=#2810 from googleapis:owl-bot-copy ddfe450 PiperOrigin-RevId: 579956341
googleapis · Nov 6, 2023 · a0103c5 · a0103c5
1 parent 7bc2e6f
commit a0103c5
Show file tree

Hide file tree

Showing 31 changed files with 455 additions and 36 deletions.
diff --git a/docs/aiplatform_v1/services_.rst b/docs/aiplatform_v1/services_.rst
@@ -0,0 +1,23 @@
+Services for Google Cloud Aiplatform v1 API
+===========================================
+.. toctree::
+    :maxdepth: 2
+
+    dataset_service
+    endpoint_service
+    featurestore_online_serving_service
+    featurestore_service
+    index_endpoint_service
+    index_service
+    job_service
+    match_service
+    metadata_service
+    migration_service
+    model_garden_service
+    model_service
+    pipeline_service
+    prediction_service
+    schedule_service
+    specialist_pool_service
+    tensorboard_service
+    vizier_service
diff --git a/docs/aiplatform_v1/types_.rst b/docs/aiplatform_v1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform v1 API
+========================================
+
+.. automodule:: google.cloud.aiplatform_v1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/aiplatform_v1beta1/services_.rst b/docs/aiplatform_v1beta1/services_.rst
@@ -0,0 +1,28 @@
+Services for Google Cloud Aiplatform v1beta1 API
+================================================
+.. toctree::
+    :maxdepth: 2
+
+    dataset_service
+    deployment_resource_pool_service
+    endpoint_service
+    feature_online_store_admin_service
+    feature_online_store_service
+    feature_registry_service
+    featurestore_online_serving_service
+    featurestore_service
+    index_endpoint_service
+    index_service
+    job_service
+    match_service
+    metadata_service
+    migration_service
+    model_garden_service
+    model_service
+    persistent_resource_service
+    pipeline_service
+    prediction_service
+    schedule_service
+    specialist_pool_service
+    tensorboard_service
+    vizier_service
diff --git a/docs/aiplatform_v1beta1/types_.rst b/docs/aiplatform_v1beta1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform v1beta1 API
+=============================================
+
+.. automodule:: google.cloud.aiplatform_v1beta1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/definition_v1/services_.rst b/docs/definition_v1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1 Schema Trainingjob Definition v1 API
+============================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/definition_v1/types_.rst b/docs/definition_v1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1 Schema Trainingjob Definition v1 API
+=========================================================================
+
+.. automodule:: google.cloud.aiplatform.v1.schema.trainingjob.definition_v1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/definition_v1beta1/services_.rst b/docs/definition_v1beta1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1beta1 Schema Trainingjob Definition v1beta1 API
+======================================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/definition_v1beta1/types_.rst b/docs/definition_v1beta1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1beta1 Schema Trainingjob Definition v1beta1 API
+===================================================================================
+
+.. automodule:: google.cloud.aiplatform.v1beta1.schema.trainingjob.definition_v1beta1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/instance_v1/services_.rst b/docs/instance_v1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1 Schema Predict Instance v1 API
+======================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/instance_v1/types_.rst b/docs/instance_v1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1 Schema Predict Instance v1 API
+===================================================================
+
+.. automodule:: google.cloud.aiplatform.v1.schema.predict.instance_v1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/instance_v1beta1/services_.rst b/docs/instance_v1beta1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1beta1 Schema Predict Instance v1beta1 API
+================================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/instance_v1beta1/types_.rst b/docs/instance_v1beta1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1beta1 Schema Predict Instance v1beta1 API
+=============================================================================
+
+.. automodule:: google.cloud.aiplatform.v1beta1.schema.predict.instance_v1beta1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/params_v1/services_.rst b/docs/params_v1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1 Schema Predict Params v1 API
+====================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/params_v1/types_.rst b/docs/params_v1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1 Schema Predict Params v1 API
+=================================================================
+
+.. automodule:: google.cloud.aiplatform.v1.schema.predict.params_v1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/params_v1beta1/services_.rst b/docs/params_v1beta1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1beta1 Schema Predict Params v1beta1 API
+==============================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/params_v1beta1/types_.rst b/docs/params_v1beta1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1beta1 Schema Predict Params v1beta1 API
+===========================================================================
+
+.. automodule:: google.cloud.aiplatform.v1beta1.schema.predict.params_v1beta1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/prediction_v1/services_.rst b/docs/prediction_v1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1 Schema Predict Prediction v1 API
+========================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/prediction_v1/types_.rst b/docs/prediction_v1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1 Schema Predict Prediction v1 API
+=====================================================================
+
+.. automodule:: google.cloud.aiplatform.v1.schema.predict.prediction_v1.types
+    :members:
+    :show-inheritance:
diff --git a/docs/prediction_v1beta1/services_.rst b/docs/prediction_v1beta1/services_.rst
@@ -0,0 +1,4 @@
+Services for Google Cloud Aiplatform V1beta1 Schema Predict Prediction v1beta1 API
+==================================================================================
+.. toctree::
+    :maxdepth: 2
diff --git a/docs/prediction_v1beta1/types_.rst b/docs/prediction_v1beta1/types_.rst
@@ -0,0 +1,6 @@
+Types for Google Cloud Aiplatform V1beta1 Schema Predict Prediction v1beta1 API
+===============================================================================
+
+.. automodule:: google.cloud.aiplatform.v1beta1.schema.predict.prediction_v1beta1.types
+    :members:
+    :show-inheritance:
diff --git a/google/cloud/aiplatform_v1/gapic_metadata.json b/google/cloud/aiplatform_v1/gapic_metadata.json
@@ -1975,6 +1975,90 @@
         }
       }
     },
+    "ScheduleService": {
+      "clients": {
+        "grpc": {
+          "libraryClient": "ScheduleServiceClient",
+          "rpcs": {
+            "CreateSchedule": {
+              "methods": [
+                "create_schedule"
+              ]
+            },
+            "DeleteSchedule": {
+              "methods": [
+                "delete_schedule"
+              ]
+            },
+            "GetSchedule": {
+              "methods": [
+                "get_schedule"
+              ]
+            },
+            "ListSchedules": {
+              "methods": [
+                "list_schedules"
+              ]
+            },
+            "PauseSchedule": {
+              "methods": [
+                "pause_schedule"
+              ]
+            },
+            "ResumeSchedule": {
+              "methods": [
+                "resume_schedule"
+              ]
+            },
+            "UpdateSchedule": {
+              "methods": [
+                "update_schedule"
+              ]
+            }
+          }
+        },
+        "grpc-async": {
+          "libraryClient": "ScheduleServiceAsyncClient",
+          "rpcs": {
+            "CreateSchedule": {
+              "methods": [
+                "create_schedule"
+              ]
+            },
+            "DeleteSchedule": {
+              "methods": [
+                "delete_schedule"
+              ]
+            },
+            "GetSchedule": {
+              "methods": [
+                "get_schedule"
+              ]
+            },
+            "ListSchedules": {
+              "methods": [
+                "list_schedules"
+              ]
+            },
+            "PauseSchedule": {
+              "methods": [
+                "pause_schedule"
+              ]
+            },
+            "ResumeSchedule": {
+              "methods": [
+                "resume_schedule"
+              ]
+            },
+            "UpdateSchedule": {
+              "methods": [
+                "update_schedule"
+              ]
+            }
+          }
+        }
+      }
+    },
     "SpecialistPoolService": {
       "clients": {
         "grpc": {

diff --git a/google/cloud/aiplatform_v1/services/migration_service/client.py b/google/cloud/aiplatform_v1/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

diff --git a/google/cloud/aiplatform_v1/types/__init__.py b/google/cloud/aiplatform_v1/types/__init__.py
@@ -559,6 +559,30 @@
 from .service_networking import (
     PrivateServiceConnectConfig,
 )
+
+from .publisher_model import (
+    PublisherModel,
+)
+from .saved_query import (
+    SavedQuery,
+)
+from .schedule import (
+    Schedule,
+)
+from .schedule_service import (
+    CreateScheduleRequest,
+    DeleteScheduleRequest,
+    GetScheduleRequest,
+    ListSchedulesRequest,
+    ListSchedulesResponse,
+    PauseScheduleRequest,
+    ResumeScheduleRequest,
+    UpdateScheduleRequest,
+)
+from .service_networking import (
+    PrivateServiceConnectConfig,
+)
+
 from .specialist_pool import (
     SpecialistPool,
 )

diff --git a/google/cloud/aiplatform_v1beta1/services/feature_registry_service/async_client.py b/google/cloud/aiplatform_v1beta1/services/feature_registry_service/async_client.py
@@ -1035,7 +1035,7 @@ async def sample_create_feature():
             response,
             self._client._transport.operations_client,
             gca_feature.Feature,
-            metadata_type=feature_registry_service.CreateRegistryFeatureOperationMetadata,
+            metadata_type=featurestore_service.CreateFeatureOperationMetadata,
         )
 
         # Done; return the response.

diff --git a/google/cloud/aiplatform_v1beta1/services/feature_registry_service/client.py b/google/cloud/aiplatform_v1beta1/services/feature_registry_service/client.py
@@ -1281,7 +1281,7 @@ def sample_create_feature():
             response,
             self._transport.operations_client,
             gca_feature.Feature,
-            metadata_type=feature_registry_service.CreateRegistryFeatureOperationMetadata,
+            metadata_type=featurestore_service.CreateFeatureOperationMetadata,
         )
 
         # Done; return the response.