o i@s^ddlZddlmZddlmZmZdgZdeegefdeeedfdee fd dZ dS) N)Iterable)AnyCallabletrace_dependenciescallableinputs.returnc sPtfdd}zt||D]}||qWtdtStdw)aTrace the execution of a callable in order to determine which modules it uses. Args: callable: The callable to execute and trace. inputs: The input to use during tracing. The modules used by 'callable' when invoked by each set of inputs are union-ed to determine all modules used by the callable for the purpooses of packaging. Returns: A list of the names of all modules used during callable execution. cs|dkrdS|jj}d}||jvr|j|j}n!||jvr$|j|j}nd|jvr9t|jd|d}|r7|jnd}|rB|dSdS)Ncallself)f_codeco_name f_globals __module__f_localsgetattradd)frameeventargnamemodulemethod modules_usede/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/torch/package/analyze/trace_dependencies.pyrecord_used_moduless   z/trace_dependencies..record_used_modulesN)setsys setprofilelist)rrrinprrrr s     ) rcollections.abcrtypingrr__all__tupler strrrrrrs