
    BVh                     X    d Z ddlZddlmZ ddlmZ  G d dej                        Zd Zy)z;Converts assert statements to their corresponding TF calls.    N)	converter)	templatesc                       e Zd ZdZd Zy)AssertTransformerzDTransforms Assert nodes to Call so they can be handled as functions.c                 d   | j                  |       d}|j                  7t        j                  ||j                  t        j                  dd             S t        |j                  t
        j                        r,t        j                  ||j                  |j                        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templates      ^/home/dcms/DCMS/lib/python3.12/site-packages/tensorflow/python/autograph/converters/asserts.pyvisit_AssertzAssertTransformer.visit_Assert   s    tH xx
yymm-D9; ; 
DHHdmm	,xdiiTXXFF KLL    N)__name__
__module____qualname____doc__r    r   r   r   r      s    LMr   r   c                 :    t        |      j                  |       } | S )N)r   visit)r   ctxs     r   	transformr   .   s    	3		%	%d	+$	+r   )	r   r    tensorflow.python.autograph.corer    tensorflow.python.autograph.pyctr   Baser   r   r   r   r   <module>r#      s+    B  6 6M	 M.r   