Skip to content

Commit 79b6541

Browse files
authored
Fix ucaip init method (#1024)
1 parent 4e3c116 commit 79b6541

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
@@ -193,12 +193,12 @@ def monkeypatch_bq(bq_client, *args, **kwargs):
193193
# aiplatform.init(args)
194194
def monkeypatch_aiplatform_init(aiplatform_klass, kaggle_kernel_credentials):
195195
aiplatform_init = aiplatform_klass.init
196-
def patched_init(self, *args, **kwargs):
196+
def patched_init(*args, **kwargs):
197197
specified_credentials = kwargs.get('credentials')
198198
if specified_credentials is None:
199199
Log.info("No credentials specified, using KaggleKernelCredentials.")
200200
kwargs['credentials'] = kaggle_kernel_credentials
201-
return aiplatform_init(self, *args, **kwargs)
201+
return aiplatform_init(*args, **kwargs)
202202

203203
if (not has_been_monkeypatched(aiplatform_klass.init)):
204204
aiplatform_klass.init = patched_init

0 commit comments

Comments
 (0)