fix(franka): reset also returns franka state

Author: juelg

extensions/rcs_fr3/src/rcs_fr3/envs.py

@@ -44,7 +44,8 @@ def _rs2dict(self, state: hw.RobotState):
     def reset(
         self, *, seed: int | None = None, options: dict[str, Any] | None = None
     ) -> tuple[dict[str, Any], dict[str, Any]]:
-        return super().reset(seed=seed, options=options)
+        obs, info = super().reset(seed=seed, options=options)
+        return self.get_obs(obs), info
 
     def close(self):
         self.hw_robot.stop_control_thread()

extensions/rcs_panda/src/rcs_panda/envs.py

@@ -44,7 +44,8 @@ def _rs2dict(self, state: hw.RobotState):
     def reset(
         self, *, seed: int | None = None, options: dict[str, Any] | None = None
     ) -> tuple[dict[str, Any], dict[str, Any]]:
-        return super().reset(seed=seed, options=options)
+        obs, info = super().reset(seed=seed, options=options)
+        return self.get_obs(obs), info
 
     def close(self):
         self.hw_robot.stop_control_thread()