Skip to content

Commit 945b9e4

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Fix failed unit tests due to google-cloud-storage upgrade.
PiperOrigin-RevId: 640313157
1 parent f9d4b51 commit 945b9e4

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

tests/unit/aiplatform/test_metadata.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -91,7 +91,10 @@
9191
_TEST_RUN = "run-1"
9292
_TEST_OTHER_RUN = "run-2"
9393
_TEST_DISPLAY_NAME = "test-display-name"
94-
_TEST_CREDENTIALS = mock.Mock(spec=credentials.AnonymousCredentials())
94+
_TEST_CREDENTIALS = mock.Mock(
95+
spec=credentials.AnonymousCredentials(),
96+
universe_domain="googleapis.com",
97+
)
9598
_TEST_BUCKET_NAME = "gs://test-bucket"
9699

97100
# resource attributes

tests/unit/aiplatform/test_metadata_models.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,10 @@
5050
_TEST_PARENT = (
5151
f"projects/{_TEST_PROJECT}/locations/{_TEST_LOCATION}/metadataStores/default"
5252
)
53-
_TEST_CREDENTIALS = mock.Mock(spec=auth_credentials.AnonymousCredentials())
53+
_TEST_CREDENTIALS = mock.Mock(
54+
spec=auth_credentials.AnonymousCredentials(),
55+
universe_domain="googleapis.com",
56+
)
5457

5558

5659
# artifact

0 commit comments

Comments
 (0)