
     Ti	                    T    S SK Jr  S SKrS SKrS SKrS SKJr        SS jrSS jr	g)    )annotationsN)helperc                  ^ [         R                  " U 5      S-  n0 m/ nUR                  S5       H  n[        UR                  [        S5      S  5      nX   n[        R                  " 5       n[        US5       nUR                  UR                  5       5        S S S 5        [        R                  R                  U5      TU'   M     [        UR                  S5      5      n	S /[        U	5      -  nUR                  S5       H  n[        UR                  [        S5      S  5      n[        R                  " 5       n
[        US5       nU
R                  UR                  5       5        S S S 5        [        R                  R                  U
5      X5'   M     [        U4S jU 5       5      (       d   S5       e[        S	 U 5       5      (       a   S
5       eTU4$ ! , (       d  f       GND= f! , (       d  f       N= f)Ntest_data_set_0z
input_*.pbinput_rbzoutput_*.pboutput_c              3  ,   >#    U  H	  oT;   v   M     g 7fN ).0nameinputss     `/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/onnxscript/utils/evaluation_utils.py	<genexpr>!load_test_data.<locals>.<genexpr>#   s     6+$v~+s   zSome inputs are missing.c              3  (   #    U  H  oS L v   M
     g 7fr   r   )r   outputs     r   r   r   $   s     A0@fT>0@s   zSome outputs are missing.)pathlibPathglobintstemlenonnxTensorProtoopenParseFromStringreadnumpy_helperto_arraylistallany)qual_model_dirinput_namestest_data_direxpected_outputs	test_dataidx
input_name
input_datafoutput_file_pathsoutput_datar   s              @r   load_test_datar0      s    LL03DDMF"''5	)..X12 %
%%'
)T"a&&qvvx0 #!..77
Cz 6 ]//>?v$5 66"''6	)..Y!123&&()T"a''1 # $ 1 1 : :; G 7 6+666R8RR6A0@AAA^C^^A###! #" #"s   9 G G(
G%	(
G6	c                   0 n[        U R                  R                  5       GHH  u  p#UR                  R                  R
                  R                   Vs/ s H  oDR                  PM     nn[        R                  " UR                  R                  R                  5      nUc,  [        SUR                  R                  R                   35      eU[        R                  [        R                  [        R                  4;   a>  [        R                   R"                  " U6 R%                  U5      S-
  XR&                  '   GM  [        R                   R)                  SSXVS9XR&                  '   GMK     U$ s  snf )zGenerate random input for the model.

NOTE: This is unused. There is parity issue with randomly generated data. Need investigation.
zUnsupported dtype: g      ?   d   )sizedtype)	enumerategraphinputtypetensor_typeshapedim	dim_valueonnx_helpertensor_dtype_to_np_dtype	elem_type
ValueErrornpfloat16float32float64randomrandastyper   randint)modelr   _r8   dr;   np_dtypes          r   generate_random_inputrN   )   s   
 Fekk//0&+jj&<&<&B&B&F&FG&F&FG77

8N8N8X8XY25::3I3I3S3S2TUVV

BJJ

;;!#!7!>!>x!H3!NF::!#!2!21c!2!VF:: 1 M Hs   E1)r%   strr&   z	list[str]returnz.tuple[dict[str, np.ndarray], list[np.ndarray]])rJ   zonnx.ModelProtorP   zdict[str, np.ndarray])

__future__r   r   numpyrB   r   r   r>   r0   rN   r       r   <module>rT      s8    #    &$$&/$3$:rS   