
    @;?i	                     f    d dl mZ d dlmZmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ  G d de      Zy	)
   )_ModelBasedConfig)
get_config
parse_args)raise_unsupported_api_error)INFER_BENCHMARK)disable_pir_bydefault)try_except_decoratorc                   &    e Zd Zd Zd Zed        Zy)Enginec                     t               }t        |j                  |j                  d      }|j                  j
                  | _        |j                  j                  | _        t        |      | _
        y )NF)	overridesshow)r   r   configoverrideGlobalmode_modeoutput_outputr   _model)selfargsr   s      d/var/www/html/leadgen/airagagent/ocr_fallback/ocr_env/lib/python3.12/site-packages/paddlex/engine.py__init__zEngine.__init__   sM    |DKK4==uM]]''
}}++'/    c                    | j                   dk(  r| j                  j                         S | j                   dk(  r%t                | j                  j	                          y| j                   dk(  r$t                | j                  j                         S | j                   dk(  r$t                | j                  j                         S | j                   dk(  r`| j                  j                         D ]B  }t        r
|j                          | j                  s'|j                  | j                         D yt        | j                    | j                         y)zthe main functioncheck_datasettrainevaluateexportpredict)	save_pathN)r   r   r   r   r   r   r    r!   r   printr   save_allr   	__class__)r   ress     r   runz
Engine.run"   s     ::(;;,,..ZZ7"!#KKZZ:%!#;;''))ZZ8#!#;;%%''ZZ9${{**, 9"		<<LL4<<L89 (4::,Hr   N)__name__
__module____qualname____doc__r   r	   r'    r   r   r   r      s     0 I Ir   r   N)modelr   utils.configr   r   utils.errorsr   utils.flagsr   utils.lazy_loaderr   utils.result_saverr	   objectr   r,   r   r   <module>r4      s'     % 0 5 ( 4 4 IV  Ir   