change summary file to json
This commit is contained in:
@@ -1,4 +1,5 @@
|
|||||||
import ast
|
import ast
|
||||||
|
import json
|
||||||
import copy
|
import copy
|
||||||
import inspect
|
import inspect
|
||||||
import math
|
import math
|
||||||
@@ -140,7 +141,7 @@ def write_to_directory(convert_func,
|
|||||||
else:
|
else:
|
||||||
raise ValueError("Directory already exists! Abort")
|
raise ValueError("Directory already exists! Abort")
|
||||||
|
|
||||||
summary_file = "dataset_summary.pkl"
|
summary_file = "dataset_summary.json"
|
||||||
|
|
||||||
metadata_recorder = {}
|
metadata_recorder = {}
|
||||||
for scenario in tqdm.tqdm(scenarios):
|
for scenario in tqdm.tqdm(scenarios):
|
||||||
@@ -178,6 +179,6 @@ def write_to_directory(convert_func,
|
|||||||
os.rename(output_path, save_path)
|
os.rename(output_path, save_path)
|
||||||
summary_file = os.path.join(save_path, summary_file)
|
summary_file = os.path.join(save_path, summary_file)
|
||||||
with open(summary_file, "wb") as 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))
|
print("Summary is saved at: {}".format(summary_file))
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user