Function process_step

eval/utils/convert_output.py:68–94

(args)

# Use multiprocessing to speed up processing
def process_step(args):
    task, episode_id, step_id, pred, base_path = args
    try:
        actions, parameters, status = parse_action(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"
        output_file_path = os.path.join(base_path, folder, file_name)

        with open(output_file_path, 'w', encoding='utf-8') as output_file:
            json.dump(transformed_entry, output_file, indent=4, ensure_ascii=False)
        
        return f"Saved transformed entry to: {output_file_path}"
    except Exception as e:
        return f"Error processing step {step_id} in episode {episode_id}: {e}"


# # Multi-threaded version