From c465de925be331b4b70ca754518866ad62f67216 Mon Sep 17 00:00:00 2001 From: Tyler Lum Date: Mon, 12 Aug 2024 21:05:16 -0700 Subject: [PATCH] Fix viewer issue where fps is not used correctly --- example/position_retargeting/hand_robot_viewer.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/example/position_retargeting/hand_robot_viewer.py b/example/position_retargeting/hand_robot_viewer.py index 6fac9ba..ecd1639 100644 --- 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()