U Mh@sZddlZddlmZmZmZmZmZdgZeegefeeedfeedddZ dS)N)AnyCallableIterableListTupletrace_dependencies.)callableinputsreturnc sHtfdd}z t||D] }||q"W5tdXtS)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|dkr dS|jj}d}||jkr0|j|j}nB||jkrH|j|j}n*d|jkrrt|jd|d}|rn|jnd}|r|dS)Ncallself)f_codeco_name f_globals __module__f_localsgetattradd)frameeventargnamemodulemethodZ modules_usedZ/var/www/html/venv/lib/python3.8/site-packages/torch/package/analyze/trace_dependencies.pyrecord_used_moduless   z/trace_dependencies..record_used_modulesN)setsys setprofilelist)rr rZinprrrrs    ) rtypingrrrrr__all__strrrrrrs