From 13749b65221e7575d9aa6831060f7bbfc1753759 Mon Sep 17 00:00:00 2001
From: QuanyiLi <quanyili0057@gmail.com>
Date: Sat, 6 May 2023 17:41:25 +0100
Subject: [PATCH] change summary file to json

---
 scenarionet/converter/utils.py                               | 5 +++--
 scenarionet/{converter/scripts => examples}/__init__.py      | 0
 .../{converter/scripts => examples}/convert_nuplan.py        | 0
 .../{converter/scripts => examples}/convert_nuscenes.py      | 0
 scenarionet/{converter/scripts => examples}/convert_waymo.py | 0
 5 files changed, 3 insertions(+), 2 deletions(-)
 rename scenarionet/{converter/scripts => examples}/__init__.py (100%)
 rename scenarionet/{converter/scripts => examples}/convert_nuplan.py (100%)
 rename scenarionet/{converter/scripts => examples}/convert_nuscenes.py (100%)
 rename scenarionet/{converter/scripts => examples}/convert_waymo.py (100%)

diff --git a/scenarionet/converter/utils.py b/scenarionet/converter/utils.py
index 982cd42..3b3db3d 100644
--- a/scenarionet/converter/utils.py
+++ b/scenarionet/converter/utils.py
@@ -1,4 +1,5 @@
 import ast
+import json
 import copy
 import inspect
 import math
@@ -140,7 +141,7 @@ def write_to_directory(convert_func,
         else:
             raise ValueError("Directory already exists! Abort")
 
-    summary_file = "dataset_summary.pkl"
+    summary_file = "dataset_summary.json"
 
     metadata_recorder = {}
     for scenario in tqdm.tqdm(scenarios):
@@ -178,6 +179,6 @@ def write_to_directory(convert_func,
     os.rename(output_path, save_path)
     summary_file = os.path.join(save_path, summary_file)
     with open(summary_file, "wb") as file:
-        pickle.dump(dict_recursive_remove_array(metadata_recorder), file)
+        json.dump(dict_recursive_remove_array(metadata_recorder), file)
     print("Summary is saved at: {}".format(summary_file))
 
diff --git a/scenarionet/converter/scripts/__init__.py b/scenarionet/examples/__init__.py
similarity index 100%
rename from scenarionet/converter/scripts/__init__.py
rename to scenarionet/examples/__init__.py
diff --git a/scenarionet/converter/scripts/convert_nuplan.py b/scenarionet/examples/convert_nuplan.py
similarity index 100%
rename from scenarionet/converter/scripts/convert_nuplan.py
rename to scenarionet/examples/convert_nuplan.py
diff --git a/scenarionet/converter/scripts/convert_nuscenes.py b/scenarionet/examples/convert_nuscenes.py
similarity index 100%
rename from scenarionet/converter/scripts/convert_nuscenes.py
rename to scenarionet/examples/convert_nuscenes.py
diff --git a/scenarionet/converter/scripts/convert_waymo.py b/scenarionet/examples/convert_waymo.py
similarity index 100%
rename from scenarionet/converter/scripts/convert_waymo.py
rename to scenarionet/examples/convert_waymo.py