o
    [Û·il  ã                   @   s`   d dl mZ d dlmZ d dlmZmZmZmZm	Z	 d dl
mZ defdd„Zdefdd	„Zd
S )é    ©ÚDynamoOptimizationResult)Úinnermost_fn)ÚListÚCallableÚDictÚUnionÚSet)ÚCodeTypeÚoriginal_codec                 C   s   t | d |ƒ}|S )Nr   )r   ÚmoduleÚresult© r   úL/home/ubuntu/vllm_env/lib/python3.10/site-packages/depyf/explain/__init__.pyÚ_extract_artifacts	   s   r   c                 C   s.   ddl m} |d sJ dƒ‚t| |ƒ}| ¡ S )Nr   )ÚdataÚis_inside_prepare_debugz5`dump_src` must be used inside `depyf.prepare_debug`.)Údepyf.explain.global_variablesr   r   Úto_src)r   r   r   Ú	artifactsr   r   r   Údump_src   s   
r   N)Údepyf.explain.utilsr   Útorch._dynamo.eval_framer   Útypingr   r   r   r   r	   Útypesr
   r   r   r   r   r   r   Ú<module>   s    