o
    i                     @   s(   d dl mZ d dlmZmZ dd ZdS )    )Gelu)KEY_TRANSFORM_REFParametricAttention_v2c                  C   s$   t t d} | tjdksJ d S )N)key_transformgelu)r   r   get_refr   name)attn r
   c/home/ubuntu/.local/lib/python3.10/site-packages/thinc/tests/layers/test_parametric_attention_v2.pytest_key_transform_used   s   r   N)thinc.layers.gelur   #thinc.layers.parametricattention_v2r   r   r   r
   r
   r
   r   <module>   s    