
    Vpf	                         d dl Z d dlZd dlZd dlmZ 	 d dlZn# e$ r dZY nw xY wd dlmZ  ej	        ej
        d          dedefd            Z ej	        ej
        d	          d
edefd            ZdS )    N)Any)profilerzpickle_util.dumpsnameobjreturnc                    t           t          d           G d dt           j                  }t          j                    5 } ||                              |            |                                cddd           S # 1 swxY w Y   dS )zBSee `pickle.dumps`. Used for serializing host callbacks in jaxlib.NNo module named "cloudpickle"c                       e Zd ZdZej        j                                        Zd eej	        <   d eej
        <   d eej        <    eed          rd eej        <   dS dS )dumps.<locals>.Picklerz'Customizes the behavior of cloudpickle.c                     | j          S )Nr   )xs    T/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/jax/_src/pickle_util.py<lambda>zdumps.<locals>.Pickler.<lambda>,   s
    16     c                     dS )NMISSING _s    r   r   zdumps.<locals>.Pickler.<lambda>-   s    ) r   c                     dS )N_HAS_DEFAULT_FACTORYr   r   s    r   r   zdumps.<locals>.Pickler.<lambda>/   s    ( r   _KW_ONLY_TYPEc                     dS )Nr   r   r   s    r   r   zdumps.<locals>.Pickler.<lambda>3   s    O r   N)__name__
__module____qualname____doc__cloudpickleCloudPicklerdispatch_tablecopydataclasses_FIELD_BASE_MISSING_TYPE_HAS_DEFAULT_FACTORY_CLASShasattrr   r   r   r   Picklerr   "   s        11 !-<AACCN /D.CN;*+0C0CN;,-(( ;9: w{O,, 
#
# [./// r   r(   )r   ModuleNotFoundErrorr    ioBytesIOdumpgetvalue)r   r(   files      r   dumpsr/      s     
=
>
>>    (   , 
z|| tGDMMs==??                 s   2BBBzpickle_util.loadsdatac                 V    t           t          d          t          j        |           S )zSee `pickle.loads`.Nr
   )r   r)   loads)r0   s    r   r2   r2   =   s*     
=
>
>>		4	 	  r   )r#   	functoolsr*   typingr   r   ImportErrorjax._srcr   partialannotate_functionbytesr/   r2   r   r   r   <module>r:      s           				         +++       8-4GHHHs u    IH@ 8-4GHHH! !# ! ! ! IH! ! !s    ##