From e4d5a0da7ebf866aa298ffadc2715b37b85fad2d Mon Sep 17 00:00:00 2001 From: bxdd Date: Thu, 30 Nov 2023 22:53:05 +0800 Subject: [PATCH] [FIX] fix client util bug --- learnware/client/utils.py | 18 ++++++------------ 1 file changed, 6 insertions(+), 12 deletions(-) diff --git a/learnware/client/utils.py b/learnware/client/utils.py index c4c5c47..049d07f 100644 --- a/learnware/client/utils.py +++ b/learnware/client/utils.py @@ -10,24 +10,18 @@ logger = get_module_logger(module_name="client_utils") def system_execute(args, timeout=None, env=None, stdout=subprocess.DEVNULL, stderr=subprocess.PIPE): - if env is None: - env = os.environ.copy() - pass - - if isinstance(args, str): - pass - else: - args = " ".join(args) - pass + env = os.environ.copy() if env is None else env + args = args if isinstance(args, str) else " ".join(args) com_process = subprocess.run(args, stdout=stdout, stderr=stderr, timeout=timeout, env=env, shell=True) try: com_process.check_returncode() except subprocess.CalledProcessError as err: - errmsg = com_process.stderr.decode() - logger.warning(f"System Execute Error: {errmsg}") - raise Exception(errmsg) + if err.stderr is not None: + errmsg = err.stderr.decode() + logger.warning(f"System Execute Error: {errmsg}") + raise err def remove_enviroment(conda_env):