Update env (#7)

* add capture script

* gymnasium API

* training with gymnasium API

scenarionet_training/scripts/local_test.py

@@ -1,6 +1,7 @@
 import os.path
 
 from metadrive.envs.scenario_env import ScenarioEnv
+from metadrive.envs.gym_wrapper import GymEnvWrapper
 
 from scenarionet import SCENARIONET_REPO_PATH, SCENARIONET_DATASET_PATH
 from scenarionet_training.train_utils.multi_worker_PPO import MultiWorkerPPO
@@ -13,8 +14,8 @@ if __name__ == '__main__':
     stop = int(100_000_000)
 
     config = dict(
-        env=env,
-        env_config=dict(
+        env=GymEnvWrapper,
+        env_config=dict(inner_class=ScenarioEnv, inner_config=dict(
             # scenario
             start_scenario_index=0,
             num_scenarios=32,
@@ -33,8 +34,7 @@ if __name__ == '__main__':
 
             # training
             horizon=None,
-            use_lateral_reward=True,
-        ),
+        )),
 
         # # ===== Evaluation =====
         evaluation_interval=2,