ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 21:33:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add better input handling for clearml-init in colab (#515)

Browse Source
This commit is contained in:
Victor Sonck 2021-12-14 19:49:34 +01:00 committed by GitHub
parent dedf31867a
commit 24464b7c10
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clearml/cli/config/__main__.py
View File

@ -67,6 +67,15 @@ def main():
print(description, end='') print(description, end='')
sentinel = '' sentinel = ''
parse_input = '' parse_input = ''
# COLAB_GPU will always be available, even when running on CPU
if os.environ.get('COLAB_GPU'):
# When running from a colab instance and calling clearml-init
# colab will squish the api credentials into a single line
# The regex splits this single line based on 2 spaces or more
import re
api_input = input()
parse_input = '\n'.join(re.split(r" {2,}", api_input))
else:
for line in iter(input, sentinel): for line in iter(input, sentinel):
parse_input += line+'\n' parse_input += line+'\n'
if line.rstrip() == '}': if line.rstrip() == '}':