format

2023-08-08 17:51:04 +01:00
parent fac13e2e5f
commit 52ff95a078
1 changed files with 35 additions and 35 deletions
--- a/scenarionet/tests/script/run_ckpt.py
+++ b/scenarionet/tests/script/run_ckpt.py
@@ -7,7 +7,8 @@ from scenarionet_training.scripts.train_nuplan import config

 if __name__ == "__main__":
    initialize_ray(test_mode=False, num_gpus=1)
-    env = createGymWrapper(ScenarioEnv)({
+    env = createGymWrapper(ScenarioEnv)(
+        {
            # "data_directory": AssetLoader.file_path("nuscenes", return_raw_style=False),
            "data_directory": "D:\\scenarionet_testset\\nuplan_test\\nuplan_test_w_raw",
            "use_render": True,
@@ -42,7 +43,6 @@ if __name__ == "__main__":
        }
    )

-
    # env.reset()
    #
    #