o
    i                  	   @   sd   d dl Z d dlmZ d dlZd dlZd dlZe jdd Zddddejdejd	e	d
e	fddZ
dS )    N)Pathc                  c   s&    t t } | V  tt|  d S )N)r   tempfilemkdtempshutilrmtreestr)d r	   S/home/ubuntu/.local/lib/python3.10/site-packages/curated_transformers/tests/util.pymake_tempdir   s   r   gh㈵>atolrtolabr   r   c                C   s   t jj| |||d d S )Nr   )torchtestingassert_close)r   r   r   r   r	   r	   r
   torch_assertclose   s   
r   )
contextlibpathlibr   r   r   r   contextmanagerr   tensorfloatr   r	   r	   r	   r
   <module>   s"    
