finshi press action

2025-09-24 11:11:24 +08:00
parent 8b89b2e35b
commit 330d903047
16 changed files with 296 additions and 91 deletions
--- a/data_gen_dependencies/action/init.py
+++ b/data_gen_dependencies/action/init.py
@@ -4,7 +4,9 @@ from .insert import InsertStage, HitStage
 from .slide import SlideStage
 from .pour import PourStage
 from .pull import PullStage
-from .push import PushStage, ClickStage
+from .push import PushStage
+from .pull_revolute import PullRevoluteStage
+from .press import PressPrismaticStage

 ACTION_STAGE = {
    "grasp": GraspStage,
@@ -19,9 +21,11 @@ ACTION_STAGE = {
    "hit": NotImplemented,
    "pour": PourStage,
    "push": PushStage,
-    'click': ClickStage,
-    'touch': ClickStage,
-    "pull": PullStage
+    'click': PressPrismaticStage,
+    'touch': PressPrismaticStage,
+    'press_prismatic': PressPrismaticStage,
+    "pull": PullStage,
+    "pull_revolute": PullRevoluteStage,
 }

 def build_stage(action):