Function execute_code_in_process

RealChart2Code_eval/evaluate_task2.py:332–364

(code_content: str, data_directory: Path, result_queue, timeout_seconds: int)

def execute_code_in_process(code_content: str, data_directory: Path, result_queue, timeout_seconds: int):
    
    try:
        import pandas as pd
        import numpy as np
        import matplotlib
        import matplotlib.pyplot as plt
        import os
        
        matplotlib.use('Agg')
        
        exec_globals = {
            '__name__': '__main__',
            'pd': pd,
            'np': np,
            'plt': plt,
            'matplotlib': matplotlib,
            'os': os,
        }
        
        try:
            import seaborn as sns
            exec_globals['sns'] = sns
        except ImportError:
            pass
        
        exec(code_content, exec_globals)
        result_queue.put(('success', 'Code execution completed'))
        
    except Exception as e:
        import traceback
        full_traceback = traceback.format_exc()
        result_queue.put(('error', f"Code execution failed: {str(e)}\n\nFull traceback:\n{full_traceback}"))


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