
    /j                     B    d dl mZ d dlZd dlmZ  G d de          ZdS )    )annotationsN)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	        |          ret          |j        d                   D ]I}|t          ||j        d         dz
                     }	|	|t	          j	        |d d |f                   |f<   JnVt          |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
             k/home/longshao/multi-rider-rag/.venv/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 	/171:&& . .SFLOa$7889*-"(1QQQT7##Q&''. 171:&& / /SFLOa$7889+.!AAAqD'W$a'((t    )NNNN)__name__
__module____qualname__r!    r"   r    r   r      s6         "!!!# # # # # #r"   r   )
__future__r   numpyr   ,onnx.reference.ops.aionnxml._op_run_aionnxmlr   r   r&   r"   r    <module>r*      sk    # " " " " "     F F F F F F$ $ $ $ $m $ $ $ $ $r"   