ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix Jupyter support on Windows host

Browse Source
This commit is contained in:
allegroai 2019-06-16 02:29:05 +03:00
parent e452709492
commit 089a4c66ff
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -22,14 +22,14 @@ class ScriptInfo(object):
@classmethod @classmethod
def _get_jupyter_notebook_filename(cls): def _get_jupyter_notebook_filename(cls):
if not sys.argv[0].endswith('/ipykernel_launcher.py') or len(sys.argv) < 3 or not sys.argv[2].endswith('.json'): if not sys.argv[0].endswith(os.path.sep+'ipykernel_launcher.py') or len(sys.argv) < 3 or not sys.argv[2].endswith('.json'):
return None return None
# we can safely assume that we can import the notebook package here # we can safely assume that we can import the notebook package here
try: try:
from notebook.notebookapp import list_running_servers from notebook.notebookapp import list_running_servers
import requests import requests
current_kernel = sys.argv[2].split('/')[-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()) server_info = next(list_running_servers())
r = requests.get( r = requests.get(
url=server_info['url'] + 'api/sessions', url=server_info['url'] + 'api/sessions',
@ -44,7 +44,7 @@ class ScriptInfo(object):
break break
notebook_path = cur_notebook['notebook']['path'] notebook_path = cur_notebook['notebook']['path']
entry_point_filename = notebook_path.split('/')[-1] entry_point_filename = notebook_path.split(os.path.sep)[-1]
# now we should try to find the actual file # now we should try to find the actual file
entry_point = (Path.cwd() / entry_point_filename).absolute() entry_point = (Path.cwd() / entry_point_filename).absolute()

2
trains/logger.py
View File

@ -75,7 +75,7 @@ class Logger(object):
self._report_worker = None self._report_worker = None
self._task_handler = None self._task_handler = None
if DevWorker.report_stdout and not PrintPatchLogger.patched: if DevWorker.report_stdout and not PrintPatchLogger.patched and not running_remotely():
Logger._stdout_proxy = PrintPatchLogger(sys.stdout, self, level=logging.INFO) Logger._stdout_proxy = PrintPatchLogger(sys.stdout, self, level=logging.INFO)
Logger._stderr_proxy = PrintPatchLogger(sys.stderr, self, level=logging.ERROR) Logger._stderr_proxy = PrintPatchLogger(sys.stderr, self, level=logging.ERROR)
self._task_handler = TaskHandler(self._task.session, self._task.id, capacity=100) self._task_handler = TaskHandler(self._task.session, self._task.id, capacity=100)