chore: internal fix

yinghsienwu · copybara-github · commit e40751908f16 · 2023-08-15T17:11:12.000-07:00
PiperOrigin-RevId: 557297465
diff --git a/google/cloud/aiplatform/constants/base.py b/google/cloud/aiplatform/constants/base.py
@@ -14,7 +14,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-
+import os
 from google.cloud.aiplatform import version as aiplatform_version
 
 
@@ -56,8 +56,8 @@
         "us-west4",
     }
 )
-
-API_BASE_PATH = "aiplatform.googleapis.com"
+# This env variable injection is for testing, but not considered to be a public API.
+API_BASE_PATH = os.environ.get("_VERTEX_API_BASE_PATH") or "aiplatform.googleapis.com"
 PREDICTION_API_BASE_PATH = API_BASE_PATH
 
 # Batch Prediction

Original file line number	Diff line number	Diff line change
`@@ -14,7 +14,7 @@`
`14`	`14`	`# See the License for the specific language governing permissions and`
`15`	`15`	`# limitations under the License.`
`16`	`16`	`#`
`17`		`-`
	`17`	`+import os`
`18`	`18`	`from google.cloud.aiplatform import version as aiplatform_version`
`19`	`19`
`20`	`20`
`@@ -56,8 +56,8 @@`
`56`	`56`	`"us-west4",`
`57`	`57`	`}`
`58`	`58`	`)`
`59`		`-`
`60`		`-API_BASE_PATH = "aiplatform.googleapis.com"`
	`59`	`+# This env variable injection is for testing, but not considered to be a public API.`
	`60`	`+API_BASE_PATH = os.environ.get("_VERTEX_API_BASE_PATH") or "aiplatform.googleapis.com"`
`61`	`61`	`PREDICTION_API_BASE_PATH = API_BASE_PATH`
`62`	`62`
`63`	`63`	`# Batch Prediction`