diff --git a/rdagent/utils/env.py b/rdagent/utils/env.py index 47d5aea2b..2f7e4dd5d 100644 --- a/rdagent/utils/env.py +++ b/rdagent/utils/env.py @@ -1522,7 +1522,7 @@ def prepare(self, *args, **kwargs) -> None: # type: ignore[no-untyped-def] qlib_data_path = next(iter(self.conf.extra_volumes.keys())) if not (Path(qlib_data_path) / "qlib_data" / "cn_data").exists(): logger.info("We are downloading!") - cmd = "python -m qlib.run.get_data qlib_data --target_dir ~/.qlib/qlib_data/cn_data --region cn --interval 1d --delete_old False" + cmd = "python -m qlib.cli.data qlib_data --name qlib_data --target_dir ~/.qlib/qlib_data/cn_data --region cn --interval 1d --delete_old False" self.check_output(entry=cmd) else: logger.info("Data already exists. Download skipped.") diff --git a/test/utils/test_env.py b/test/utils/test_env.py index a550b4d4a..ab76b00eb 100644 --- a/test/utils/test_env.py +++ b/test/utils/test_env.py @@ -23,7 +23,7 @@ class QlibLocalEnv(LocalEnv): def prepare(self) -> None: if not (Path("~/.qlib/qlib_data/cn_data").expanduser().resolve().exists()): self.check_output( - entry="python -m qlib.run.get_data qlib_data --target_dir ~/.qlib/qlib_data/cn_data --region cn", + entry="python -m qlib.cli.data qlib_data --name qlib_data --target_dir ~/.qlib/qlib_data/cn_data --region cn", ) else: print("Data already exists. Download skipped.")