U md @s4dZddlZddlZz ddlZWnek r8dZYnXddlZejfddZddZ e dkr0ddl Z e j dd Z e jd d d d e jdd dd e jdedddddddgddejdeddejdeddejddddd gd!dejd"ejdejdddgeddd#gZe ee dS)$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).0Zeachr]/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/joblib/test/data/create_numpy_pickle.py sz&get_joblib_version...cSsg|]}|dk r|dqS)N)group)rmrrrrs)splitjoin)joblib_versionmatchesrrrget_joblib_versions rc Cs i}|j}|j}t}dtj}dtj ddd}|rL|dkrLdnd}|r|dkrjd|d<d } n|d f|d<d |} |j rd |d <|d7}nd} d||||| } zt j || f|Wn4t k r} ztd| || W5d} ~ XYnXtd| dS)Nz {0[0]}{0[1]}rzlibZ _compressedTcompressz.gzz.pkl.{}r cache_sizeZ _cache_sizez.pklzjoblib_{}{}_pickle_py{}_np{}{}zCError: cannot generate file '{}' with arguments '{}'. Error was: {}z!File '{}' generated successfully.)rmethodrformatsys version_infor np __version__r rjoblibdump Exceptionprint) to_pickleargskwargsrrr py_versionZ numpy_versionbody extensionZpickle_filenameerrrwrite_test_pickle!sB    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--methodrgzipbz2xzlzmaZlz4zSet compression method.)typedefaultchoicesr-zs<  %