o
    "jw                     @   s<   d dl mZ d dlmZ ddlmZ g ZG dd deZdS )    )ast_to_source_code)gast   )BaseTransformerc                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	AssertTransformerz=
    A class transforms python assert to convert_assert.
    c                 C   s
   || _ d S N)root)selfr    r
   h/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/dy2static/assert_transformer.py__init__   s   
zAssertTransformer.__init__c                 C   s   |  | j d S r   )visitr   )r	   r
   r
   r   	transform   s   zAssertTransformer.transformc                 C   s>   t djt|j|jrt|jnddjd j}t j|dS )Nz_jst.Assert({test}, {msg}) )testmsgr   )value)	r   parseformatr   r   r   bodyr   Expr)r	   nodeZconvert_assert_noder
   r
   r   visit_Assert"   s   zAssertTransformer.visit_AssertN)__name__
__module____qualname____doc__r   r   r   r
   r
   r
   r   r      s
    r   N)Zpaddle.jit.dy2static.utilsr   Zpaddle.utilsr   Zbase_transformerr   __all__r   r
   r
   r
   r   <module>   s
   