o
     ¢iÒ	  ã                   @   sŠ   d dl Z d dlmZ d dlmZ e j dg d¢¡dd„ ƒZe j dg d¢¡d	d
„ ƒZe j dg d¢¡dd„ ƒZ	e j ddg¡dd„ ƒZ
dS )é    N)ÚregistryÚ
model_name)
z/spacy-curated-transformers.AlbertTransformer.v1z-spacy-curated-transformers.BertTransformer.v1z2spacy-curated-transformers.CamembertTransformer.v1z0spacy-curated-transformers.RobertaTransformer.v1z-spacy-curated-transformers.XlmrTransformer.v1z.spacy-curated-transformers.WithStridedSpans.v1z*spacy-curated-transformers.ScalarWeight.v1z7spacy-curated-transformers.TransformerLayersListener.v1z:spacy-curated-transformers.LastTransformerLayerListener.v1z5spacy-curated-transformers.ScalarWeightingListener.v1c                 C   ó   t j | ¡ d S ©N©Úspacy_registryÚarchitecturesÚget)r   © r
   úb/home/ubuntu/.local/lib/python3.10/site-packages/spacy_curated_transformers/tests/test_registry.pyÚtest_model_from_registry   s   r   Úencoder_name)z2spacy-curated-transformers.BertWordpieceEncoder.v1z,spacy-curated-transformers.ByteBpeEncoder.v1z;spacy-curated-transformers.CamembertSentencepieceEncoder.v1z)spacy-curated-transformers.CharEncoder.v1z2spacy-curated-transformers.SentencepieceEncoder.v1z.spacy-curated-transformers.WordpieceEncoder.v1z6spacy-curated-transformers.XlmrSentencepieceEncoder.v1c                 C   s   t j | ¡ƒ  d S r   r   )r   r
   r
   r   Ú$test_tokenizer_encoder_from_registry   s   r   Úloader_name)z+spacy-curated-transformers.ByteBpeLoader.v1z/spacy-curated-transformers.CharEncoderLoader.v1z8spacy-curated-transformers.HFTransformerEncoderLoader.v1z2spacy-curated-transformers.HFPieceEncoderLoader.v1z5spacy-curated-transformers.PyTorchCheckpointLoader.v1z1spacy-curated-transformers.SentencepieceLoader.v1z-spacy-curated-transformers.WordpieceLoader.v1c                 C   r   r   )r   Úmodel_loadersr	   )r   r
   r
   r   Ú test_model_loaders_from_registry+   s   r   Úcallback_namez<spacy-curated-transformers.gradual_transformer_unfreezing.v1c                 C   r   r   )r   Ú	callbacksr	   )r   r
   r
   r   Útest_callbacks_from_registry<   s   r   )ÚpytestÚ
spacy.utilr   r   Úspacy_curated_transformers.utilÚmarkÚparametrizer   r   r   r   r
   r
   r
   r   Ú<module>   s0    þ
þ
þ
ÿþ