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)