[update] update default sapien example task

yzqin · yzqin · commit 72408d7b5299 · 2023-08-25T21:39:37.000-07:00
diff --git a/example/sapien/run_sapien_viz.py b/example/sapien/run_sapien_viz.py
@@ -64,7 +64,7 @@ def wrapped_setup_viewer(self):
 control_mode = ["base_pd_joint_vel_arm_pd_joint_vel"] * 3 + ["pd_joint_delta_pos"] * 8
 
 # You can try different task_num to visualize different tasks
-task_num = 1
+task_num = 7
 
 env = gym.make(task_names[task_num], obs_mode="rgbd", control_mode=control_mode[task_num])
 # print("Observation space", env.observation_space)
diff --git a/setup.py b/setup.py
@@ -4,7 +4,6 @@
 
 """Setup script """
 
-import os
 import re
 from pathlib import Path
 
@@ -21,15 +20,6 @@
     else:
         raise RuntimeError("Unable to find __version__ string.")
 
-
-def collect_files(target_dir):
-    file_list = []
-    for (root, dirs, files) in os.walk(target_dir, followlinks=True):
-        for filename in files:
-            file_list.append(os.path.join('..', root, filename))
-    return file_list
-
-
 core_requirements = [
     "numpy<=1.23.0",
     "numpy-quaternion",
@@ -100,7 +90,7 @@ def setup_package():
         license_files=("LICENSE",),
         packages=packages,
         package_data={
-            "sim_web_visualizer": package_files
+            name: package_files
         },
         python_requires='>=3.6,<3.11',
         zip_safe=True,
