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

Fix poetry support inside virtualenv with pyenv

Browse Source
This commit is contained in:
allegroai 2020-01-26 15:05:59 +02:00
parent a57a5b151c
commit 2afaff1713
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
trains_agent/helper/package/poetry_api.py
View File

@ -59,8 +59,15 @@ class PoetryConfig:
func = kwargs.pop("func", Argv.get_output) func = kwargs.pop("func", Argv.get_output)
kwargs.setdefault("stdin", DEVNULL) kwargs.setdefault("stdin", DEVNULL)
kwargs['env'] = deepcopy(os.environ) kwargs['env'] = deepcopy(os.environ)
if 'VIRTUAL_ENV' in kwargs['env'] or 'CONDA_PREFIX' in kwargs['env']:
kwargs['env'].pop('VIRTUAL_ENV', None) kwargs['env'].pop('VIRTUAL_ENV', None)
kwargs['env'].pop('CONDA_PREFIX', None) kwargs['env'].pop('CONDA_PREFIX', None)
kwargs['env'].pop('PYTHONPATH', None)
if hasattr(sys, "real_prefix") and hasattr(sys, "base_prefix"):
path = ':'+kwargs['env']['PATH']
path = path.replace(':'+sys.base_prefix, ':'+sys.real_prefix, 1)
kwargs['env']['PATH'] = path
if check_if_command_exists("poetry"): if check_if_command_exists("poetry"):
argv = Argv("poetry", *args) argv = Argv("poetry", *args)
else: else: