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

Support limiting pip version, limit to <20 by default

Browse Source
This commit is contained in:
allegroai
2020-01-22 12:02:12 +02:00
parent ae2775f7b8
commit 284271c654
4 changed files with 28 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
trains_agent/helper/package/base.py
View File

@@ -17,6 +17,7 @@ class PackageManager(object):
_selected_manager = None
_cwd = None
_pip_version = None
@abc.abstractproperty
def bin(self):
@@ -65,7 +66,7 @@ class PackageManager(object):
pass
def upgrade_pip(self):
return self._install("pip", "--upgrade")
return self._install("pip"+self.get_pip_version(), "--upgrade")
def get_python_command(self, extra=()):
# type: (...) -> Executable
@@ -123,3 +124,17 @@ class PackageManager(object):
except Exception:
pass
return []
@classmethod
def set_pip_version(cls, version):
if not version:
return
version = version.replace(' ', '')
if ('=' in version) or ('~' in version) or ('<' in version) or ('>' in version):
cls._pip_version = version
else:
cls._pip_version = "=="+version
@classmethod
def get_pip_version(cls):
return cls._pip_version or ''