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

Add optional ignored packages in script requirements (currently used for pywin32)

Browse Source
This commit is contained in:
allegroai 2021-05-31 15:44:11 +03:00
parent 206d86ec5b
commit 0454ffcdf4
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -27,6 +27,7 @@ class ScriptInfoError(Exception):
class ScriptRequirements(object): class ScriptRequirements(object):
_max_requirements_size = 512 * 1024 _max_requirements_size = 512 * 1024
_packages_remove_version = ('setuptools', ) _packages_remove_version = ('setuptools', )
_ignore_packages = ('pywin32',)
def __init__(self, root_folder): def __init__(self, root_folder):
self._root_folder = root_folder self._root_folder = root_folder
@ -170,6 +171,8 @@ class ScriptRequirements(object):
# requirement summary # requirement summary
requirements_txt += '\n' requirements_txt += '\n'
for k, v in reqs.sorted_items(): for k, v in reqs.sorted_items():
if k.lower() in ScriptRequirements._ignore_packages:
continue
version = v.version version = v.version
if k in forced_packages: if k in forced_packages:
forced_version = forced_packages.pop(k, None) forced_version = forced_packages.pop(k, None)