B 8ƨfd@sBddlZddlZddlZddlZddlmZdddZddZdS) N)wrapsFcsfdd}|S)Ncstfdd}|S)Ncsltj}|jj}tj|\}}r8t|ddSt dt j d||\}}t|||dS)Nz is skip.iO4)inspect currentframef_backf_code co_filenameospathsplitextprinttorch manual_seednprandomseed generate_data)argskwargsframe file_path case_path_inputsoutputs)fnskipW/data/nfs/dataset/workspace/mindspore_dataset/mindspore-tests-benchmark/common/utils.pywrapper s   z6auto_generate_data..decorator..wrapper)r)rr)r)rr decorator sz%auto_generate_data..decoratorr)rrr)rrauto_generate_datas r cCst|tstdt|t|ts8tdt|t|tsTtdt|x4t|D](\}}|dt|d}t||q^Wx4t|D](\}}|dt|d}t||qWdS)Nzcase_path must be str, but got zinputs must be list, but got zoutputs must be list, but got _inputz.npy_output) isinstancestr ValueErrortypelist enumeratersave)rrridxdata file_namerrrr!s   r)F) rrr numpyr functoolsrr rrrrrs