U
    ‰dã  ã                   @   s@   d dl Z d dlZd dlZd dlZd dlZeje dœdd„ZdS )é    N)Úreturnc                  C   sp   t  ¡ tj t  ¡ g t  ¡ g tj tj tj	¡¡g } t
dd„ | D ƒƒ}tdd„ |D ƒƒsbt‚dd„ |D ƒS )Nc                 S   s   h | ]}t j |¡’qS © )ÚosÚpathÚabspath©Ú.0Úir   r   ú@/tmp/pip-unpacked-wheel-ua33x9lu/torch/profiler/python_tracer.pyÚ	<setcomp>   s     z _prefix_regex.<locals>.<setcomp>c                 s   s   | ]}t |tƒV  qd S )N)Ú
isinstanceÚstrr   r   r   r
   Ú	<genexpr>   s     z _prefix_regex.<locals>.<genexpr>c                 S   s   g | ]}|t j ‘qS r   )r   Úsepr   r   r   r
   Ú
<listcomp>   s     z!_prefix_regex.<locals>.<listcomp>)ÚsiteÚgetsitepackagesÚsysr   ÚgetuserbaseÚgetusersitepackagesr   ÚdirnameÚtorchÚ__file__ÚsortedÚallÚAssertionError)Z	raw_pathsZpath_prefixesr   r   r
   Ú_prefix_regex	   s    ÿþýüÿr   )r   r   r   Útypingr   ÚListr   r   r   r   r   r
   Ú<module>   s
   