Skip to content

Commit ef0e8b9

Browse files
committed
Fixed ucaip integration to work with latest code
1 parent 8abf69a commit ef0e8b9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

patches/kaggle_gcp.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -198,7 +198,7 @@ def patched_init(self, *args, **kwargs):
198198
if specified_credentials is None:
199199
Log.info("No credentials specified, using KaggleKernelCredentials.")
200200
kwargs['credentials'] = kaggle_kernel_credentials
201-
return client_init(self, *args, **kwargs)
201+
return client_init(self, *args, **kwargs)
202202

203203
if (not has_been_monkeypatched(client_klass.init)):
204204
client_klass.init = patched_init
@@ -340,7 +340,7 @@ def init_ucaip():
340340
kaggle_kernel_credentials = KaggleKernelCredentials(target=GcpTarget.CLOUDAI)
341341

342342
# Patch the ucaip init method, this flows down to all ucaip services
343-
monkeypatch_init(aiplatform.initializer.global_config, kaggle_kernel_credentials)
343+
monkeypatch_init(aiplatform, kaggle_kernel_credentials)
344344

345345
def init_video_intelligence():
346346
from google.cloud import videointelligence

0 commit comments

Comments
 (0)