ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix Task.add_requirements() passing package_version starting with @ or ; or #

Browse Source
This commit is contained in:
allegroai 2021-05-19 15:24:09 +03:00
parent 83fb00ef8b
commit 8fb4fb2b5e
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -216,10 +216,10 @@ class ScriptRequirements(object):
requirements_txt += '{0}\n'.format(version)
elif k.startswith('-e '):
requirements_txt += '{0} {1}\n'.format(k.replace('-e ', '', 1), version or '')
elif version and (str(version).strip() or ' ')[0] in '><~=':
requirements_txt += '{0} {1}\n'.format(k, version)
elif version and str(version).strip():
elif version and str(version or ' ').strip()[0].isdigit():
requirements_txt += '{0} {1} {2}\n'.format(k, '==', version)
elif version and str(version).strip():
requirements_txt += '{0} {1}\n'.format(k, version)
else:
requirements_txt += '{0}\n'.format(k)
return requirements_txt

3
clearml/logger.py
View File

@ -1014,7 +1014,8 @@ class Logger(object):
figure=figure,
iter=iteration or 0,
logger=self,
force_save_as_image=False if report_interactive else ('png' if report_image else True),
force_save_as_image=False if report_interactive and not report_image
else ('png' if report_image else True),
)
def set_default_upload_destination(self, uri):