o
    i                     @   sR   d dl mZ d dlZd dlZedgZdd Zdd Zdd	 Zej	d
d Z
dS )    )PathNcpuc                 C   s,   z| j dddd W d S  ty   Y d S w )Nz--slow
store_truezinclude slow tests)actionhelp)	addoption
ValueError)parser r
   W/home/ubuntu/.local/lib/python3.10/site-packages/curated_transformers/tests/conftest.pypytest_addoption	   s
   r   c                 C   s   |  dd d S )Nmarkerszslow: include slow tests)addinivalue_line)configr
   r
   r   pytest_configure   s   r   c                    s>    fdd}dD ]}| j v r||std| d qd S )Nc                    s    j d|  dS )Nz--F)r   	getoption)optitemr
   r   getopt   s   z$pytest_runtest_setup.<locals>.getopt)slowzneed --z option to run)keywordspytestskip)r   r   r   r
   r   r   pytest_runtest_setup   s   
r   c                 C   s   t | j t| jjS )N)printfspathr   parent)requestr
   r
   r   test_dir&   s   
r   )pathlibr   r   torchdeviceTORCH_DEVICESr   r   r   fixturer   r
   r
   r
   r   <module>   s    	