diff --git a/robot_sf/gym_env/pedestrian_env.py b/robot_sf/gym_env/pedestrian_env.py index 3de32be..72c1617 100644 --- a/robot_sf/gym_env/pedestrian_env.py +++ b/robot_sf/gym_env/pedestrian_env.py @@ -324,7 +324,6 @@ def save_recording(self, filename: str = None): # only save if there are recorded states if len(self.recorded_states) == 0: logger.warning("No states recorded, skipping save") - # TODO: First env.reset will always have no recorded states return os.makedirs(os.path.dirname(filename), exist_ok=True) diff --git a/robot_sf/render/sim_view.py b/robot_sf/render/sim_view.py index 691d22d..606d974 100644 --- a/robot_sf/render/sim_view.py +++ b/robot_sf/render/sim_view.py @@ -92,7 +92,7 @@ class SimulationView: map_def: MapDefinition = field(default_factory=MapDefinition) obstacles: List[Obstacle] = field(default_factory=list) caption: str = "RobotSF Simulation" - focus_on_robot: bool = False + focus_on_robot: bool = True focus_on_ego_ped: bool = False record_video: bool = False video_path: str = None