File size: 723 Bytes
866f598 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
import zipfile
import json
# Need to add option flags to specify the file path to the agent
archive = zipfile.ZipFile("dqn_v2-5/ALE-Pacman-v5.zip", "r")
file = archive.open("data")
byte_file = file.read()
json_file = json.loads(byte_file.decode("utf-8"))
# Only want to remove serialized objects from dictionary
val_to_remove = ":serialized:"
for key in json_file.keys():
# So if each value is a type dict, then I want to iterate through it and remove the serialized key
if type(json_file[key]) is dict:
if val_to_remove in json_file[key].keys():
json_file[key].pop(val_to_remove)
outfile = open("configtest.json", "w")
json.dump(json_file, outfile, indent=2,)
file.close()
outfile.close()
|