o
    toi                     @  s4   d dl mZ d dlZd dlmZ G dd deZdS )    )annotationsN)OpRunAiOnnxMlc                   @  s.   e Zd Z												dddZdS )LabelEncoderNc                   s   |p|p|p|}|	p|
p|p|}t t||dd ||u r%| |j}n$||	u r/|tj}n||
u r9|tj}n||u rI|tj}tt	sIdtj
 fdd|gd}||}|jtkre|tj}|fS )NF)strict c                   s     | S )N)get)xclassesdefval `/home/ubuntu/.local/lib/python3.10/site-packages/onnx/reference/ops/aionnxml/op_label_encoder.py<lambda>.   s    z#LabelEncoder._run.<locals>.<lambda>)otypes)dictzipitemdtypenpfloat32int64str_
isinstancestr	vectorizeobjectastype)selfr   default_floatdefault_int64default_stringdefault_tensorkeys_floatskeys_int64skeys_stringsvalues_floatsvalues_int64svalues_stringskeys_tensorvalues_tensorkeysvaluesotypelookup_funcoutputr   r	   r   _run   s,   

zLabelEncoder._run)NNNNNNNNNNNN)__name__
__module____qualname__r/   r   r   r   r   r      s    r   )
__future__r   numpyr   ,onnx.reference.ops.aionnxml._op_run_aionnxmlr   r   r   r   r   r   <module>   s   