From 7c4244ccc0bcff57aec6419a78f4ad8800990a95 Mon Sep 17 00:00:00 2001 From: Junjia Liu Date: Tue, 15 Oct 2024 13:21:50 +0800 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20Fix=20some=20bugs?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- rofunc/utils/oslab/path.py | 5 ++--- setup.py | 4 +++- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/rofunc/utils/oslab/path.py b/rofunc/utils/oslab/path.py index cb4ef94e..19af748a 100644 --- a/rofunc/utils/oslab/path.py +++ b/rofunc/utils/oslab/path.py @@ -35,7 +35,6 @@ def get_rofunc_path(extra_path=None): return rofunc_path - def get_elegantrl_path(): """ Get the path of the elegantrl package. @@ -84,6 +83,6 @@ def check_package_exist(package_name: str): except ImportError: package_name = package_name.replace("_", "-") - rf.logger.beauty_print(f"{package_name} is not installed, installing {package_name}...", type="warning") + print(f"{package_name} is not installed, installing {package_name}...") subprocess.check_call([sys.executable, "-m", "pip", "install", package_name]) - rf.logger.beauty_print(f"{package_name} is installed successfully!") + print(f"{package_name} is installed successfully!") diff --git a/setup.py b/setup.py index 35022ba6..29bc03ce 100644 --- a/setup.py +++ b/setup.py @@ -47,7 +47,9 @@ 'networkx', 'dgl', 'trimesh==4.0.5', - 'wandb==0.16.2'], + 'wandb==0.16.2', + 'mujoco', + 'lxml'], python_requires=">=3.7,<3.11", keywords=['robotics', 'robot learning', 'learning from demonstration', 'reinforcement learning', 'robot manipulation'],