o
    ?e                     @   sB   d Z ddlZddlmZ ddlmZ G dd dejZdd ZdS )	z;Converts assert statements to their corresponding TF calls.    N)	converter)	templatesc                   @   s   e Zd ZdZdd ZdS )AssertTransformerzDTransforms Assert nodes to Call so they can be handled as functions.c                 C   s^   |  | d}|jd u rtj||jtjdd ddS t|jtjr+tj||j|jdS td)Nz/
      ag__.assert_stmt(test, lambda: msg)
    zAssertion error)kind)testmsgz)can only convert string messages for now.)	generic_visitr   r   replacer   gastConstant
isinstanceNotImplementedError)selfnodetemplate r   o/home/www/facesmatcher.com/pyenv/lib/python3.10/site-packages/tensorflow/python/autograph/converters/asserts.pyvisit_Assert   s   

zAssertTransformer.visit_AssertN)__name__
__module____qualname____doc__r   r   r   r   r   r      s    r   c                 C   s   t || } | S )N)r   visit)r   ctxr   r   r   	transform.   s   r   )	r   r
   Z tensorflow.python.autograph.corer   Z tensorflow.python.autograph.pyctr   ZBaser   r   r   r   r   r   <module>   s   