g dZddlZddlZ ddlZddlZejfdZdZ e dk(r!ddl Z e jdZ e jdd d e jd d d e jdedgddej dej"dej dej"dej$dddddgdej dej&j)ej*gdej"ddgZe ee j/yy#e$rdZYHwxYw)zP This script is used to generate test data for joblib/test/test_numpy_pickle.py Nc|jdDcgc]}tjd|}}dj|Dcgc]}||j dc}Scc}wcc}w)zNormalize joblib version by removing suffix. >>> get_joblib_version('0.8.4') '0.8.4' >>> get_joblib_version('0.8.4b1') '0.8.4' >>> get_joblib_version('0.9.dev0') '0.9' .z(\d+).*)splitrematchjoingroup)joblib_versioneachmatchesms Y/var/www/openai/venv/lib/python3.12/site-packages/joblib/test/data/create_numpy_pickle.pyget_joblib_versionrsl*//4644xx D)4 6 88BAAMQWWQZB CC6BsA$A) A)cti}|j}|j}t}djtj }dj tjjddd}|r|dk(rdnd}|r<|dk(rd|d<d } n|d f|d<d j|} |jr d |d <|dz }nd} dj||||| } tj|| fi|tdj| y#t$r&} tdj| || Yd} ~ yd} ~ wwxYw)Nz {0[0]}{0[1]}rzlib _compressedTcompressz.gzz.pkl.{}r cache_size _cache_sizez.pklzjoblib_{}{}_pickle_py{}_np{}{}z!File '{}' generated successfully.zCError: cannot generate file '{}' with arguments '{}'. Error was: {})rmethodrformatsys version_infor np __version__rrjoblibdumpprint Exception) to_pickleargskwargsrrr py_version numpy_versionbody extensionpickle_filenamees rwrite_test_pickler-!sH F}}H [[F')N&&s'7'78JGGBNN005bq9:M&&F*:=D V !%F: I"(!F: !((0I ??#$F< M !D 6==j-DOK I9&9 188IJ B $f_fa@ B BBsD D7D22D7__main__zJoblib pickle data generator.) descriptionz --cache_size store_truez;Force creation of companion numpy files for pickled arrays.)actionhelpz --compresszGenerate compress pickles.z--methodr)rgzipbz2xzlzmalz4zSet compression method.)typedefaultchoicesr2zrSsv '-&8&8 D"KJ z $X $ $2>?F |9:  \9;  f L681HBHHUO41HBHHUO41e11%56cB3bhh/7799HBHHUO<')Ii!2!2!453u BsEEE