26 lines
895 B
Python
26 lines
895 B
Python
import json
|
|
import os
|
|
import sys
|
|
|
|
def generate_output(input_config, time):
|
|
with open(input_config, "r") as f:
|
|
data = json.load(f)
|
|
output_dir_with_taskid = os.path.join(data["output_dir"], str(data["task_id"]))
|
|
# Ensure the output directory exists before writing the output file
|
|
os.makedirs(output_dir_with_taskid, exist_ok=True)
|
|
output_data = {
|
|
"task_id": data["task_id"],
|
|
"convert_time": time,
|
|
"data_dir": data["data_dir"],
|
|
"output_dir": output_dir_with_taskid,
|
|
"episode_num": data["episode_num"],
|
|
"no_language": data["no_language"],
|
|
}
|
|
output_json_path = os.path.join(output_dir_with_taskid, "output.json")
|
|
with open(output_json_path, "w") as f:
|
|
json.dump(output_data, f)
|
|
|
|
if __name__ == "__main__":
|
|
input_config = sys.argv[1]
|
|
time = int(sys.argv[2])
|
|
generate_output(input_config, time) |