chore: expose model registry class (#1719)

googleapis · Oct 7, 2022 · dde9ba1 · dde9ba1
1 parent cba7fbf
commit dde9ba1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
@@ -46,6 +46,7 @@
 from google.cloud.aiplatform.models import Endpoint
 from google.cloud.aiplatform.models import PrivateEndpoint
 from google.cloud.aiplatform.models import Model
+from google.cloud.aiplatform.models import ModelRegistry
 from google.cloud.aiplatform.model_evaluation import ModelEvaluation
 from google.cloud.aiplatform.jobs import (
     BatchPredictionJob,
@@ -147,6 +148,7 @@
     "ImageDataset",
     "HyperparameterTuningJob",
     "Model",
+    "ModelRegistry",
     "ModelEvaluation",
     "ModelDeploymentMonitoringJob",
     "PipelineJob",