From fdfd5764821ab14c249aeeaeeb9c5a5fc4efe185 Mon Sep 17 00:00:00 2001 From: GeneLiuXe <356340460@qq.com> Date: Sun, 3 Dec 2023 15:32:13 +0800 Subject: [PATCH] [MNT] modify logger details --- learnware/client/container.py | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/learnware/client/container.py b/learnware/client/container.py index d94b5a6..65973a1 100644 --- a/learnware/client/container.py +++ b/learnware/client/container.py @@ -243,7 +243,7 @@ class ModelDockerContainer(ModelContainer): "-m", "pip", "install", - "learnware[full]", + "learnware", ] ) ) @@ -339,14 +339,10 @@ class ModelDockerContainer(ModelContainer): "--conda-env", f"{conda_env}", ] - ), - stream=True + ) ) - for output in result.output: - print(output.decode("utf-8")) - if result.exit_code != 0: - logger.error(f"Install environment dependencies in docker failed!") + logger.error(f"Install environment dependencies in docker failed!\n{result.output.decode('utf-8')}") # run_cmd_times = 10 # with tempfile.TemporaryDirectory(prefix="learnware_") as tempdir: