o
    hY                  	   @   sv   d dl Z d dlmZ dedefddZ	ddee d	ee d
edefddZ	ddee d	ee d
edefddZdS )    N)Optional	reductionreturnc                 C   s\   | dkrd}|S | dkrd}|S | dkrt d d}|S | dkr%d}|S d	}t|  d
)Nnoner   mean   elementwise_meanzPreduction='elementwise_mean' is deprecated. Please use reduction='mean' instead.sum   z# is not a valid value for reduction)warningswarn
ValueError)r   ret r   g/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/torch/nn/_reduction.pyget_enum   s"   r   Tsize_averagereduceemit_warningc                 C   sP   d}| d u rd} |d u rd}| r|rd}n|rd}nd}|r&t || |S )NzSsize_average and reduce args will be deprecated, please use reduction='{}' instead.Tr   r	   r   )r   r   format)r   r   r   warningr   r   r   r   legacy_get_string    s   r   c                 C   s   t t| ||S )N)r   r   )r   r   r   r   r   r   legacy_get_enum7   s   r   )T)	r   typingr   strintr   boolr   r   r   r   r   r   <module>   s.    
