
    e!h              
          d Z ddlZddlZ	 ddlZddlZej                  fdZd Z	e
dk(  r!ddlZ ej                  d      Zej                  dd	d
       ej                  dd	d       ej                  dedg dd        ej                   d ej"                  d             ej                   d ej"                  d             ej$                  dddddgd       ej                   dej&                        j)                          ej*                  g d ej"                  d            dgZ e	eej/                                yy# e$ r dZY Hw xY w)zP
This script is used to generate test data for joblib/test/test_numpy_pickle.py
    Nc                     | j                  d      D cg c]  }t        j                  d|       }}dj                  |D cg c]  }||j	                  d       c}      S c c}w c c}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+).*   )splitrematchjoingroup)joblib_versioneachmatchesms       l/var/www/html/diagnosisapp-backend/venv/lib/python3.12/site-packages/joblib/test/data/create_numpy_pickle.pyget_joblib_versionr      s]     7E6J6J36OPdrxx
D)PGP88BAAMQWWQZBCC QBs   A$A)
A)c                 t   i }|j                   }|j                  }t               }dj                  t        j
                        }dj                  t        j                  j                  d      d d       }|r|dk(  rdnd}|r<|dk(  rd|d<   d	}	n|d
f|d<   dj                  |      }	|j                  rd|d<   |dz  }nd}	dj                  |||||	      }
	 t        j                  | |
fi | t        dj                  |
             y # t        $ r&}t        dj                  |
||             Y d }~y d }~ww xY w)Nz{0[0]}{0[1]} r      zlib_compressedTcompressz.gz   z.pkl.{}r   
cache_size_cache_sizez.pklzjoblib_{}{}_pickle_py{}_np{}{}z!File '{}' generated successfully.zCError: cannot generate file '{}' with arguments '{}'. Error was: {})r   methodr   formatsysversion_infor	   np__version__r   r   joblibdumpprint	Exception)	to_pickleargskwargsr   r   r   
py_versionnumpy_versionbody	extensionpickle_filenamees               r   write_test_pickler-       sC   F}}H[[F')N&&s'7'78JGGBNN005bq9:M &&F*:=DV!%F:I"(!F:!((0I??#$F< M!D	6==j-O
KI9&9 	188IJ  
 	"F?FA>	
 	

s   D 	D7D22D7__main__zJoblib pickle data generator.)descriptionz--cache_size
store_truez;Force creation of companion numpy files for pickled arrays.)actionhelpz
--compresszGenerate compress pickles.z--methodr   )r   gzipbz2xzlzmalz4zSet compression method.)typedefaultchoicesr2      z<i8)dtypez<f8r   abcr   )abO   )r   r   r   u   C'est l'été !)__doc__r   r   numpyr   ImportErrorr    r   r   r-   __name__argparseArgumentParserparseradd_argumentstraranger<   arrayuint8tobytesmatrixr$   
parse_args     r   <module>rS      sy   
 
  '-&8&8 D%KP z$X$$1PQF
J  
 \0L   <&   			!8288E?+		!8288E?+!U!!,-S9		#RXX&..0		)8288E?3	I i!2!2!45E y  	Bs   E EE