Merge pull request #34 from tylerlum/2024-08-12_FixViewerFPS

[fix] Fix viewer issue in the sapien viewer where fps is not used correctly
dexsuite · Aug 13, 2024 · ebb0d58 · ebb0d58
2 parents eca0f0b + c465de9
commit ebb0d58
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/example/position_retargeting/hand_robot_viewer.py b/example/position_retargeting/hand_robot_viewer.py
@@ -163,11 +163,11 @@ def render_dexycb_data(self, data: Dict, fps=5, y_offset=0.8):
                 rgb = (np.clip(rgb, 0, 1) * 255).astype(np.uint8)
                 writer.write(rgb[..., ::-1])
             else:
-                for k in range(start_frame):
+                for _ in range(step_per_frame):
                     self.viewer.render()
 
-        if self.headless:
-            writer.release()
-        else:
+        if not self.headless:
             self.viewer.paused = True
             self.viewer.render()
+        else:
+            writer.release()