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()