
    
Ng                     6    d dl Zd dlmZ  G d de          ZdS )    N)OpRunAiOnnxMlc                       e Zd Z	 	 	 	 ddZdS )ImputerNc                    |t          |          dk    r|}|}n)|t          |          dk    r|}|}nt          d          t          |t                    rt	          j        |          }t          |j                  dk    rt          d|j                   |j        d         |j        d         dfvr+t          d|j        d          d|j        d                    |                                }t	          j	        |          rft          d|j        d                   D ]I}|t          ||j        d         dz
                     }	|	|t	          j	        |d d |f                   |f<   JnWt          d|j        d                   D ];}|t          ||j        d         dz
                     }	|	||d d |f         |k    |f<   <|fS )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
             b/var/www/html/ai-engine/env/lib/python3.11/site-packages/onnx/reference/ops/aionnxml/op_imputer.py_runzImputer._run   s     +4H0I0IA0M0M)F*GG!-#6J2K2Ka2O2O)F*GG7888fd## 	&Xf%%Fqw<<1HqwHHIII<?171:q/11Gfl1oGG171:GG   FFHH8G 	/1agaj)) . .SFLOa$7889*-"(1QQQT7##Q&''. 1agaj)) / /SFLOa$7889+.!AAAqD'W$a'((t    )NNNN)__name__
__module____qualname__r     r!   r   r   r      s6         "!!!# # # # # #r!   r   )numpyr   ,onnx.reference.ops.aionnxml._op_run_aionnxmlr   r   r%   r!   r   <module>r(      sY        F F F F F F$ $ $ $ $m $ $ $ $ $r!   