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

Support poetry when agent is installed inside virtualenv

Browse Source
This commit is contained in:
allegroai 2020-01-22 11:22:43 +02:00
parent eb012f5c24
commit ae2775f7b8
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,7 +1,9 @@
from copy import deepcopy
from functools import wraps from functools import wraps
import attr import attr
import sys import sys
import os
from pathlib2 import Path from pathlib2 import Path
from trains_agent.helper.process import Argv, DEVNULL, check_if_command_exists from trains_agent.helper.process import Argv, DEVNULL, check_if_command_exists
from trains_agent.session import Session, POETRY from trains_agent.session import Session, POETRY
@ -56,6 +58,9 @@ class PoetryConfig:
def run(self, *args, **kwargs): def run(self, *args, **kwargs):
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'].pop('VIRTUAL_ENV', None)
kwargs['env'].pop('CONDA_PREFIX', None)
if check_if_command_exists("poetry"): if check_if_command_exists("poetry"):
argv = Argv("poetry", *args) argv = Argv("poetry", *args)
else: else: