From 7bc2d83e75609aea3b8b2beca3c36be0bdf84976 Mon Sep 17 00:00:00 2001 From: Xu Yang Date: Mon, 15 Jul 2024 08:31:27 +0000 Subject: [PATCH] remove useless print command --- test/utils/test_env.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/test/utils/test_env.py b/test/utils/test_env.py index f108d320..79c7a901 100644 --- a/test/utils/test_env.py +++ b/test/utils/test_env.py @@ -2,10 +2,11 @@ import os import sys import unittest from pathlib import Path + sys.path.append(str(Path(__file__).resolve().parent.parent)) -from rdagent.utils.env import QTDockerEnv, LocalEnv, LocalConf import shutil +from rdagent.utils.env import LocalConf, LocalEnv, QTDockerEnv DIRNAME = Path(__file__).absolute().resolve().parent @@ -30,9 +31,9 @@ class EnvUtils(unittest.TestCase): ) qle = LocalEnv(conf=local_conf) qle.prepare() - conf_path = str(DIRNAME / "env_tpl" / "conf.yaml") + conf_path = str(DIRNAME / "env_tpl" / "conf.yaml") qle.run(entry="qrun " + conf_path) - mlrun_p = DIRNAME / "env_tpl" / "mlruns" + mlrun_p = DIRNAME / "env_tpl" / "mlruns" self.assertTrue(mlrun_p.exists(), f"Expected output file {mlrun_p} not found") def test_docker(self): @@ -41,17 +42,15 @@ class EnvUtils(unittest.TestCase): And run the docker image with `qrun conf.yaml` """ qtde = QTDockerEnv() - qtde.prepare() qtde.prepare() # you can prepare for multiple times. It is expected to handle it correctly # the stdout are returned as result result = qtde.run(local_path=str(DIRNAME / "env_tpl"), entry="qrun conf.yaml") - - mlrun_p = DIRNAME / "env_tpl" / "mlruns" + + mlrun_p = DIRNAME / "env_tpl" / "mlruns" self.assertTrue(mlrun_p.exists(), f"Expected output file {mlrun_p} not found") # read experiment result = qtde.run(local_path=str(DIRNAME / "env_tpl"), entry="python read_exp_res.py") - print("here") print(result)