zIf dZddlZddlZ ddlZn #e$rdZYnwxYwddlZejfdZdZ e dkr"ddl Z e j dZ e dd d e d d d e dedgddejdejdejdejdejdddddgdejdejejgdejddgZe ee dSdS)zP This script is used to generate test data for joblib/test/test_numpy_pickle.py Nc~d|dD}dd|DS)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' c8g|]}tjd|S)z(\d+).*)rematch).0eachs X/var/www/piapp/venv/lib/python3.11/site-packages/joblib/test/data/create_numpy_pickle.py z&get_joblib_version..s4666x D))666.c<g|]}||dS)N)group)rms r r z&get_joblib_version..s#BBBAAMQWWQZZMMMr )splitjoin)joblib_versionmatchess r get_joblib_versionrsO66)//44666G 88BBBBB C CCr ci}|j}|j}t}dtj}dtj ddd}|r|dkrdnd}|r<|dkrd|d<d } n|d f|d<d |} |j r d |d <|dz }nd} d||||| } tj || fi|td| dS#t$r/} td| || Yd} ~ dSd} ~ wwxYw)Nz {0[0]}{0[1]}r zlib _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_infornp __version__rrjoblibdumpprint Exception) to_pickleargskwargsrrr py_version numpy_versionbody extensionpickle_filenamees r write_test_pickler2!s F}H [F'))N&&s'788JGGBN0055bqb9::M& C&F*:*:==D  V  !%F: II"(!F: !((00I ? "#$F< M !D 6==j-DDOK I99&999 188IIJJJJJ BBB $f_fa@@ B B B B B B B B BBsD E$E  E__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.)typedefaultchoicesr7zrWs&  BBB '-&8 D D D D"K"K"KJ zOOO $X $2>???F |9:::  \9;;;  f L L L68881HBHUOO4441HBHUOO4441e11%5%56cBBB3bh///7799999HBHUOO<<<')Ii!2!2!4!4555553s