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'],