o
    i                     @   sz   d dl Z d dlmZ d dlZejdd Zejjddejdddd	d
Z	ejjddejdddddZ
dS )    N)Pathc                 C   s
  t d t d dd l}dd l}ddlm} t|  s d S tj	
dd}t| }|j|dd}W d    n1 s=w   Y  |d	|d
}d|_||dddii dd |jD }|D ]#}	|	d D ]}
|
jdkr|
jD ]}t| qot|
j d|
j qeq_d S )N	nbconvertnbformatr   )ExecutePreprocessorNOTEBOOK_KERNELpython3   )
as_versioniX  )timeoutkernel_nameTmetadatapath/c                 S   s   g | ]}d |v r|qS )outputs ).0cr   r   M/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/test_examples.py
<listcomp>   s    ztest_files.<locals>.<listcomp>r   errorz: )pytestimportorskipr   r   nbconvert.preprocessorsr   r   existsosenvirongetopenreadallow_errors
preprocesscellsoutput_type	tracebackprint	Exceptionenameevalue)nb_filer   r   r   r   fnbproccells_with_outputscelloutputlr   r   r   
test_files   s0   





r0   zNotebooks should be updated)reasonr(   )z0examples/01_intro_model_definition_methods.ipynbz%examples/05_benchmarking_layers.ipynbc                 C      d S Nr   r0   r   r   r   test_ipython_notebooks    s   	r5   z1these notebooks need special software or hardware)z examples/00_intro_to_thinc.ipynbz*examples/02_transformers_tagger_bert.ipynbz&examples/03_pos_tagger_basic_cnn.ipynbz*examples/03_textcat_basic_neural_bow.ipynbz&examples/04_configure_gpu_memory.ipynbz'examples/04_parallel_training_ray.ipynbz$examples/05_visualizing_models.ipynbz'examples/06_predicting_like_terms.ipynbc                 C   r2   r3   r   r4   r   r   r   test_ipython_notebooks_slow,   s   r6   )r0   N)r   pathlibr   r   fixturer0   markskipparametrizer5   r6   r   r   r   r   <module>   s"    
