
    h                     6    S r SSKJr  SSKJr   " S S\5      rg)z 
Implements ONNX's backend API.
    )
BackendRep)
RunOptionsc                   $    \ rS rSrSrS rS rSrg)OnnxRuntimeBackendRep   zf
Computes the prediction for a pipeline converted into
an :class:`onnxruntime.InferenceSession` node.
c                     Xl         g)z7
:param session: :class:`onnxruntime.InferenceSession`
N_session)selfsessions     Y/var/www/fran/franai/venv/lib/python3.13/site-packages/onnxruntime/backend/backend_rep.py__init__OnnxRuntimeBackendRep.__init__   s	          c                    [        5       nUR                  5        H#  u  pE[        X45      (       d  M  [        X4U5        M%     [	        U[
        5      (       a  0 n[        U R                  R                  5       5       H  u  pxX   XhR                  '   M     U R                  R                  SXc5      n	[	        U	[
        5      (       a  U	$ U R                  R                  5        V
s/ s H  oR                  PM     nn
U Vs/ s H  oU   PM	     sn$ U R                  R                  5       n[        U5      S:w  a  [        S[        U5       S35      eUS   R                  U0nU R                  R                  SXc5      $ s  sn
f s  snf )zH
Computes the prediction.
See :meth:`onnxruntime.InferenceSession.run`.
N   zModel expect z inputsr   )r   itemshasattrsetattr
isinstancelist	enumerater
   
get_inputsnamerunget_outputslenRuntimeError)r   inputskwargsoptionskvinpsiinpoutsooutput_namesr   s                r   r   OnnxRuntimeBackendRep.run   s=    ,LLNDAw""A& # fd##D#DMM$<$<$>?!'XX @==$$T49D$%%040I0I0KL0K10KL/;<|tT
|<<--**,C3x1}"]3s8*G#DEEFKK(D==$$T499  M<s   $E>?Fr	   N)__name__
__module____qualname____firstlineno____doc__r   r   __static_attributes__ r   r   r   r      s    
 :r   r   N)r/   onnx.backend.baser   onnxruntimer   r   r1   r   r   <module>r4      s   
 ) "&:J &:r   