ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-01-31 09:07:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make sure trains is included in Jupyter requirements

Browse Source
This commit is contained in:
allegroai 2020-04-26 23:11:51 +03:00
parent dbb3346332
commit d9f2f01bbd
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -223,6 +223,13 @@ class _JupyterObserver(object):
last_update_ts = None
counter = 0
prev_script_hash = None
try:
from ....version import __version__
our_module = cls.__module__.split('.')[0], __version__
except:
our_module = None
# main observer loop, check if we need to exit
while not cls._exit_event.wait(timeout=0.):
# wait for timeout or sync event
@ -266,6 +273,10 @@ class _JupyterObserver(object):
fmodules, _ = file_import_modules(notebook.parts[-1], script_code)
fmodules = ScriptRequirements.add_trains_used_packages(fmodules)
installed_pkgs = get_installed_pkgs_detail()
# make sure we are in installed packages
if our_module and (our_module[0] not in installed_pkgs):
installed_pkgs[our_module[0]] = our_module
reqs = ReqsModules()
for name in fmodules:
if name in installed_pkgs: