From 514031fce90b6e4606279d4903dc93b0f18b9f2a Mon Sep 17 00:00:00 2001
From: sasha-gitg <44654632+sasha-gitg@users.noreply.github.com>
Date: Fri, 29 Oct 2021 13:20:36 -0400
Subject: [PATCH] fix: use version.py for versioning (#804)

---
 google/cloud/aiplatform/__init__.py | 12 +++++++++---
 google/cloud/aiplatform/version.py  | 18 ++++++++++++++++++
 setup.py                            |  8 ++++++--
 3 files changed, 33 insertions(+), 5 deletions(-)
 create mode 100644 google/cloud/aiplatform/version.py

diff --git a/google/cloud/aiplatform/__init__.py b/google/cloud/aiplatform/__init__.py
index 17c6952f35..a07296378e 100644
--- a/google/cloud/aiplatform/__init__.py
+++ b/google/cloud/aiplatform/__init__.py
@@ -15,10 +15,14 @@
 # limitations under the License.
 #
 
-from google.cloud.aiplatform import gapic
-from google.cloud.aiplatform import explain
+
+from google.cloud.aiplatform import version as aiplatform_version
+
+__version__ = aiplatform_version.__version__
+
 
 from google.cloud.aiplatform import initializer
+
 from google.cloud.aiplatform.datasets import (
     ImageDataset,
     TabularDataset,
@@ -26,16 +30,18 @@
     TimeSeriesDataset,
     VideoDataset,
 )
+from google.cloud.aiplatform import explain
+from google.cloud.aiplatform import gapic
 from google.cloud.aiplatform import hyperparameter_tuning
 from google.cloud.aiplatform.metadata import metadata
 from google.cloud.aiplatform.models import Endpoint
 from google.cloud.aiplatform.models import Model
-from google.cloud.aiplatform.pipeline_jobs import PipelineJob
 from google.cloud.aiplatform.jobs import (
     BatchPredictionJob,
     CustomJob,
     HyperparameterTuningJob,
 )
+from google.cloud.aiplatform.pipeline_jobs import PipelineJob
 from google.cloud.aiplatform.tensorboard import Tensorboard
 from google.cloud.aiplatform.training_jobs import (
     CustomTrainingJob,
diff --git a/google/cloud/aiplatform/version.py b/google/cloud/aiplatform/version.py
new file mode 100644
index 0000000000..c010e08d0a
--- /dev/null
+++ b/google/cloud/aiplatform/version.py
@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+
+# Copyright 2021 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+#
+
+__version__ = "1.6.1"
diff --git a/setup.py b/setup.py
index 24e48aa745..e1602d288a 100644
--- a/setup.py
+++ b/setup.py
@@ -21,14 +21,18 @@
 import setuptools  # type: ignore
 
 name = "google-cloud-aiplatform"
-version = "1.6.1"
-description = "Cloud AI Platform API client library"
+description = "Vertex AI API client library"
 
 package_root = os.path.abspath(os.path.dirname(__file__))
 readme_filename = os.path.join(package_root, "README.rst")
 with io.open(readme_filename, encoding="utf-8") as readme_file:
     readme = readme_file.read()
 
+version = {}
+with open(os.path.join(package_root, "google/cloud/aiplatform/version.py")) as fp:
+    exec(fp.read(), version)
+version = version["__version__"]
+
 tensorboard_extra_require = ["tensorflow >=2.3.0, <=2.5.0"]
 metadata_extra_require = ["pandas >= 1.0.0"]
 xai_extra_require = ["tensorflow >=2.3.0, <=2.5.0"]