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

Add upload HTML preview of jupyter notebook as an artifact

Browse Source
This commit is contained in:
allegroai 2021-02-01 23:37:44 +02:00
parent ebc0733357
commit e3109c96d8
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
clearml/backend_interface/task/repo/scriptinfo.py
View File

@ -2,7 +2,7 @@ import os
import sys
from copy import copy
from functools import partial
from tempfile import mkstemp
from tempfile import mkstemp, gettempdir
import attr
import logging
@ -378,6 +378,18 @@ class _JupyterObserver(object):
task.upload_artifact(
name='notebook', artifact_object=Path(local_jupyter_filename),
preview='No preview available')
# noinspection PyBroadException
try:
from nbconvert.exporters import HTMLExporter # noqa
html, _ = HTMLExporter().from_filename(filename=local_jupyter_filename)
local_html = Path(gettempdir()) / 'notebook_{}.html'.format(task.id)
with open(local_html.as_posix(), 'wt') as f:
f.write(html)
task.upload_artifact(
name='notebook preview', artifact_object=local_html,
preview=' ', delete_after_upload=True)
except Exception:
pass
current_script_hash = hash(script_code + (current_cell or ''))
if prev_script_hash and prev_script_hash == current_script_hash: