🐛 Fix some bugs

Skylark0924 · Oct 15, 2024 · 7c4244c · 7c4244c
1 parent 3031a95
commit 7c4244c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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'],