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

Support broken Jupyter version on some conda installations (SageMaker)

Browse Source
This commit is contained in:
allegroai 2020-01-10 13:33:19 +02:00
parent 80f3dc6790
commit f65ef3e757
1 changed files with 32 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
trains/backend_interface/task/repo/scriptinfo.py
View File

@ -326,10 +326,38 @@ class ScriptInfo(object):
from notebook.notebookapp import list_running_servers from notebook.notebookapp import list_running_servers
import requests import requests
current_kernel = sys.argv[2].split(os.path.sep)[-1].replace('kernel-', '').replace('.json', '') current_kernel = sys.argv[2].split(os.path.sep)[-1].replace('kernel-', '').replace('.json', '')
server_info = next(list_running_servers()) try:
r = requests.get( server_info = next(list_running_servers())
url=server_info['url'] + 'api/sessions', except Exception:
headers={'Authorization': 'token {}'.format(server_info.get('token', '')), }) # on some jupyter notebook versions this function can crash on parsing the json file,
# we will parse it manually here
import ipykernel
from glob import glob
import json
for f in glob(os.path.join(os.path.dirname(ipykernel.get_connection_file()), 'nbserver-*.json')):
try:
with open(f, 'r') as json_data:
server_info = json.load(json_data)
except:
server_info = None
if server_info:
break
try:
r = requests.get(
url=server_info['url'] + 'api/sessions',
headers={'Authorization': 'token {}'.format(server_info.get('token', '')), })
except requests.exceptions.SSLError:
# disable SSL check warning
from urllib3.exceptions import InsecureRequestWarning
requests.packages.urllib3.disable_warnings(category=InsecureRequestWarning)
# fire request
r = requests.get(
url=server_info['url'] + 'api/sessions',
headers={'Authorization': 'token {}'.format(server_info.get('token', '')), }, verify=False)
# enable SSL check warning
import warnings
warnings.simplefilter('default', InsecureRequestWarning)
r.raise_for_status() r.raise_for_status()
notebooks = r.json() notebooks = r.json()