fix key error in summary dict

README.md

@@ -169,6 +169,7 @@ pyright src/agent
 ## TODO
 
 - [x] Address CI errors
-- [ ] Add coding guidelines
+- [x] Add coding guidelines
 - [x] Add teleop interface and examples
 - [x] Add BC algorithms
+- [ ] Add more examples

examples/run_ppo_gs.py

@@ -75,7 +75,7 @@ def main(num_envs: int = 2048, show_viewer: bool = False, train: bool = True) ->
 
         print("Training completed successfully!")
         print(f"Training completed in {train_summary_info['total_time']:.2f} seconds.")
-        print(f"Total episodes: {train_summary_info['total_episodes']}.")
+        print(f"Total iterations: {train_summary_info['total_iterations']}.")
         print(f"Total steps: {train_summary_info['total_steps']}.")
         print(f"Total reward: {train_summary_info['final_reward']:.2f}.")
     else:

examples/run_ppo_gym.py

@@ -107,7 +107,7 @@ def main(train: bool = True) -> None:
         train_summary_info = runner.train(metric_logger=logger)
         print("Training completed successfully!")
         print(f"Training completed in {train_summary_info['total_time']:.2f} seconds.")
-        print(f"Total episodes: {train_summary_info['total_episodes']}.")
+        print(f"Total iterations: {train_summary_info['total_iterations']}.")
         print(f"Total steps: {train_summary_info['total_steps']}.")
         print(f"Total reward: {train_summary_info['final_reward']:.2f}.")