Skip to content

Commit fef5e4d

Browse files
Ark-kuncopybara-github
authored andcommitted
chore: Platform - Transferred the ownership of the vertexai.tuning module to the Tuning team
PiperOrigin-RevId: 667915562
1 parent 2711c17 commit fef5e4d

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/unit/vertexai/test_tuning.py renamed to tests/unit/vertexai/tuning/test_tuning.py

+12
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@
2424
from unittest import mock
2525
import uuid
2626

27+
from google import auth
28+
from google.auth import credentials as auth_credentials
2729
from google.cloud import aiplatform
2830
import vertexai
2931
from google.cloud.aiplatform import compat
@@ -52,6 +54,16 @@
5254
_TEST_LOCATION = "us-central1"
5355

5456

57+
@pytest.fixture(scope="module")
58+
def google_auth_mock():
59+
with mock.patch.object(auth, "default") as auth_mock:
60+
auth_mock.return_value = (
61+
auth_credentials.AnonymousCredentials(),
62+
"test-project",
63+
)
64+
yield auth_mock
65+
66+
5567
_global_tuning_jobs: Dict[str, gca_tuning_job.TuningJob] = {}
5668

5769

0 commit comments

Comments
 (0)