Function convert2aitz_single_thread

eval/utils/convert_output.py:127–162

(input_path, output_path)

# # Single-threaded version
def convert2aitz_single_thread(input_path, output_path):
    data = load_json_data(input_path)    
    base_path = os.path.join(output_path)

    for item in data:
        task = item.get("category", "unknown")
        episode_id = item.get("episode_id", "unknown")
        steps = item.get("steps", [item])

        for index, each_step in enumerate(steps):
            step_id = index if "steps" in item else each_step.get("step_id", index)

            actions, parameters, status = parse_action(each_step["pred"])

            transformed_entry = {
                "action_predict": {
                    "COA": {
                        "txt": {
                            "ACTION": actions,
                            "ARGS": parameters,
                            "STATUS": status
                        },
                    }
                }
            }
            folder = f"{task}-{episode_id}"
            file_name = f"{folder}_{step_id}.json"
            folder_path = os.path.join(base_path, folder)
            output_path = os.path.join(folder_path, file_name)

            os.makedirs(folder_path, exist_ok=True)

            with open(output_path, 'w', encoding='utf-8') as output_file:
                json.dump(transformed_entry, output_file, indent=4, ensure_ascii=False)

            print(f"Saved transformed entry to: {output_path}")