Function execute_code_with_timeout

RealChart2Code_eval/evaluate_task2.py:367–392

(code_content: str, data_directory: Path, timeout_seconds: int = 120)

def execute_code_with_timeout(code_content: str, data_directory: Path, timeout_seconds: int = 120):
    
    result_queue = Queue()
    
    process = Process(target=execute_code_in_process, 
                     args=(code_content, data_directory, result_queue, timeout_seconds))
    
    process.start()
    process.join(timeout=timeout_seconds)
    
    if process.is_alive():
        process.terminate()
        process.join(timeout=5)
        
        if process.is_alive():
            process.kill()
            process.join()
        
        raise ExecutionTimeoutError(f"Code execution timeout (exceeded {timeout_seconds} seconds)")
    
    if not result_queue.empty():
        result_type, result_message = result_queue.get()
        if result_type == 'error':
            raise Exception(result_message)
    else:
        raise Exception("Code execution terminated abnormally")


def execute_and_save_plot(code: str, task_info: Dict, results_dir: Path, timeout: int) -> Tuple[bool, str, Optional[Path]]: