Skip to content

Conversation

@rosbo
Copy link
Contributor

@rosbo rosbo commented Jul 1, 2020

Import tensorflow_gcs_config only when set_tensorflow_credential is called to prevent preloading TensorFlow.

BUG=159906185

@rosbo rosbo requested a review from vimota July 2, 2020 00:08
@rosbo
Copy link
Contributor Author

rosbo commented Jul 2, 2020

I will test manually in Kaggle notebook to make sure the following works:

from kaggle_secrets import UserSecretsClient
user_secrets = UserSecretsClient()
user_credential = user_secrets.get_gcloud_credential()

user_secrets.set_tensorflow_credential(user_credential)

Testing in progress...

@rosbo
Copy link
Contributor Author

rosbo commented Jul 2, 2020

I confirm it works.

Copy link
Contributor

@vimota vimota left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@rosbo rosbo merged commit a901aa3 into master Jul 2, 2020
@rosbo rosbo deleted the fix-tensorflow-import branch July 2, 2020 00:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants