Skip to content

Commit 1675650

Browse files
committed
Better comments and variable names
1 parent ef0e8b9 commit 1675650

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

patches/kaggle_gcp.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -187,22 +187,22 @@ def monkeypatch_bq(bq_client, *args, **kwargs):
187187
bq_client, *args, **kwargs)
188188
return bigquery
189189

190-
# Monkey patch classes that use the init method
190+
# Monkey patch for aiplatform init
191191
# eg
192192
# from google.cloud import aiplatform
193193
# aiplatform.init(args)
194-
def monkeypatch_init(client_klass, kaggle_kernel_credentials):
195-
client_init = client_klass.init
194+
def monkeypatch_aiploatform_init(aiplatform_klass, kaggle_kernel_credentials):
195+
aiplatform_init = aiplatform_klass.init
196196
def patched_init(self, *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 client_init(self, *args, **kwargs)
201+
return aiplatform_init(self, *args, **kwargs)
202202

203-
if (not has_been_monkeypatched(client_klass.init)):
204-
client_klass.init = patched_init
205-
Log.info(f"Client patched: {client_klass}")
203+
if (not has_been_monkeypatched(aiplatform_klass.init)):
204+
aiplatform_klass.init = patched_init
205+
Log.info("aiplatform.init patched")
206206

207207
def monkeypatch_client(client_klass, kaggle_kernel_credentials):
208208
client_init = client_klass.__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, kaggle_kernel_credentials)
343+
monkeypatch_aiplatform_init(aiplatform, kaggle_kernel_credentials)
344344

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

0 commit comments

Comments
 (0)