o
    toi                     @  s@   d dl mZ d dlZd dlZd dlmZ dZG dd deZdS )    )annotationsN)OpRun)UOc                   @  s   e Zd ZdddZdS )RegexFullMatchNc              
     sz   |j jtvrtd|j  zt| W n tjy* } ztd||d }~ww tj	 fddtj
gd}||fS )Nz,Input must be string tensor, received dtype zInvalid regex pattern c                   s     | d uS N)	fullmatch)xregex Z/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/op_regex_full_match.py<lambda>"   s    z%RegexFullMatch._run.<locals>.<lambda>)otypes)dtypekind_acceptable_str_dtypes	TypeErrorrecompileerror
ValueErrornp	vectorizebool_)selfr	   patternefullmatch_funcr   r
   r   _run   s   

zRegexFullMatch._runr   )__name__
__module____qualname__r   r   r   r   r   r      s    r   )	
__future__r   r   numpyr   onnx.reference.op_runr   r   r   r   r   r   r   <module>   s   