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

Fix jupyter notebook packages and uncommitted changes are sometimes not fetched

Browse Source
This commit is contained in:
allegroai 2024-07-09 15:35:19 +03:00
parent 253aee3b0e
commit 0bb0f29867
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -563,8 +563,15 @@ class _JupyterObserver(object):
reqs = ReqsModules()
for name in fmodules:
if name in installed_pkgs:
pkg_name, version = installed_pkgs[name]
reqs.add(pkg_name, version, fmodules[name])
# handle namespace packages, which are returned as flat dicts of format
# {mapping_pkg_name: (pkg_name, version), ...}
if isinstance(installed_pkgs[name], dict):
for subpackage_name, subpackage in installed_pkgs[name].items():
pkg_name, version = subpackage
reqs.add(pkg_name, version, fmodules.get(subpackage_name, fmodules[name]))
else:
pkg_name, version = installed_pkgs[name]
reqs.add(pkg_name, version, fmodules[name])
requirements_txt, conda_requirements = ScriptRequirements.create_requirements_txt(reqs)
# remove ipython direct access from the script code