We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2711c17 commit fef5e4dCopy full SHA for fef5e4d
tests/unit/vertexai/test_tuning.py renamed to tests/unit/vertexai/tuning/test_tuning.py
@@ -24,6 +24,8 @@
24
from unittest import mock
25
import uuid
26
27
+from google import auth
28
+from google.auth import credentials as auth_credentials
29
from google.cloud import aiplatform
30
import vertexai
31
from google.cloud.aiplatform import compat
@@ -52,6 +54,16 @@
52
54
_TEST_LOCATION = "us-central1"
53
55
56
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
67
_global_tuning_jobs: Dict[str, gca_tuning_job.TuningJob] = {}
68
69
0 commit comments