o lh  @s*dZddlZddlZzddlZWn eydZYnwddlZejfddZddZ e dkrddl Z e j dd Z e jd d d d e jdd dd e jdedgdddejdeddejdeddejdddddgddejdejdejgdeddd gZe ee dSdS)!zP This script is used to generate test data for joblib/test/test_numpy_pickle.py NcCs(dd|dD}ddd|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' cSsg|]}td|qS)z(\d+).*)rematch).0eachrV/var/www/html/env/lib/python3.10/site-packages/joblib/test/data/create_numpy_pickle.py sz&get_joblib_version...cSsg|] }|dur|dqS)N)group)rmrrrrs)splitjoin)joblib_versionmatchesrrrget_joblib_versions rc Csi}|j}|j}t}dtj}dtj ddd}|r&|dkr&dnd}|rL|dkr5d|d<d } n |d f|d<d |} |j rKd |d <|d7}nd} d||||| } z t j || fi|Wnt y} ztd| || WYd} ~ dSd} ~ wwtd| dS)Nz {0[0]}{0[1]}r zlib _compressedTcompressz.gzz.pkl.{}r cache_size _cache_sizez.pklzjoblib_{}{}_pickle_py{}_np{}{}zCError: cannot generate file '{}' with arguments '{}'. Error was: {}z File '{}' generated successfuly.)rmethodrformatsys version_infornp __version__r rjoblibdump Exceptionprint) to_pickleargskwargsrrr py_version numpy_versionbody extensionpickle_filenameerrrwrite_test_pickle!s<     r-__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.)typedefaultchoicesr2zsB  $