(aiplatform_klass, kaggle_kernel_credentials)
| 191 | # from google.cloud import aiplatform |
| 192 | # aiplatform.init(args) |
| 193 | def monkeypatch_aiplatform_init(aiplatform_klass, kaggle_kernel_credentials): |
| 194 | aiplatform_init = aiplatform_klass.init |
| 195 | def patched_init(*args, **kwargs): |
| 196 | specified_credentials = kwargs.get('credentials') |
| 197 | if specified_credentials is None: |
| 198 | logging.info("No credentials specified, using KaggleKernelCredentials.") |
| 199 | kwargs['credentials'] = kaggle_kernel_credentials |
| 200 | return aiplatform_init(*args, **kwargs) |
| 201 | |
| 202 | if (not has_been_monkeypatched(aiplatform_klass.init)): |
| 203 | aiplatform_klass.init = patched_init |
| 204 | logging.info("aiplatform.init patched") |
| 205 | |
| 206 | def monkeypatch_client(client_klass, kaggle_kernel_credentials): |
| 207 | client_init = client_klass.__init__ |
no test coverage detected