
    Si                     :    S SK Jr  S SKrS SKJr   " S S\5      rg)    )annotationsN)OpRunAiOnnxMlc                  &    \ rS rSr    SS jrSrg)Imputer   Nc                ~   Ub  [        U5      S:  a  UnUnO"Ub  [        U5      S:  a  UnUnO[        S5      e[        U[        5      (       a  [        R
                  " U5      n[        UR                  5      S:w  a  [        SUR                   35      eUR                  S   UR                  S   S4;  a+  [        SUR                  S    SUR                  S    35      eUR                  5       n[        R                  " U5      (       aa  [        UR                  S   5       HB  nU[        XR                  S   S-
  5         n	X[        R                  " US S 2U4   5      U4'   MD     U4$ [        UR                  S   5       H1  nU[        XR                  S   S-
  5         n	XUS S 2U4   U:H  U4'   M3     U4$ )Nr   zMissing are not defined.   z x must be a matrix but shape is    zDimension mismatch z != )len
ValueError
isinstancelistnparrayshape	TypeErrorcopyisnanrangemin)
selfximputed_value_floatsimputed_value_int64sreplaced_value_floatreplaced_value_int64valuesreplaceivals
             e/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnx/reference/ops/aionnxml/op_imputer.py_runImputer._run   s     +4H0IA0M)F*G!-#6J2Ka2O)F*G788fd##XXf%Fqww<1>qwwiHII<<?1771:q/1%fll1o%6d1771:,G  FFH88G1771:&SLLOa$789*-"((1QT7#Q&' ' t	 1771:&SLLOa$789+.!AqD'W$a'( ' t     )NNNN)__name__
__module____qualname____firstlineno__r"   __static_attributes__r%   r$   r!   r   r      s     "!!!#r$   r   )
__future__r   numpyr   ,onnx.reference.ops.aionnxml._op_run_aionnxmlr   r   r%   r$   r!   <module>r.      s    #  F$m $r$   