a
    dD                     @  s  d Z ddlmZ ddlZddlZddlZddlZddlm	Z	 ddl
ZddlmZ ddlmZ ddlmZ ddlmZmZmZ dd	lmZ e	rdd
lmZ dd Zdddddddddddddddddddddddddddd d!ZG d"d# d#ejZeeje_eejejd < d$d%d&d'Z e ejej!d < d$d%d(d)Z"ze"ejej#d < W n e$e%fyf   Y n0 d<d+d,d-d.d/Z&d0d1d2d3d4d,d5d5d6d7d8Z'ej(d9d%d:d;Z)dS )=z/
Support pre-0.12 series pickle compatibility.
    )annotationsN)TYPE_CHECKING)NDArrayBacked)
BaseOffset)Index)DatetimeArrayPeriodArrayTimedeltaArray)BlockManager)	Generatorc                 C  s   | j }| }|d }z|| |d< W d S  ty } zd}|t|v rz&|d }t||d< W W Y d }~d S  ty   Y n0 n~|rt|d trt|d t	r|d }|j| |d< W Y d }~d S |rt|d t
r|d }tj| |d< W Y d }~d S  W Y d }~n
d }~0 0 d S )Nz:_reconstruct: First argument must be a sub-type of ndarrayr   )stackpop	TypeErrorstrobject__new__
isinstancetype
issubclassr   r   r   )selfr   argsfuncerrmsgcls r   d/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/pandas/compat/pickle_compat.pyload_reduce   s0     r   )pandas.core.arraysSparseArray)numpyZndarray)pandas.core.indexes.frozen
FrozenList)pandas.core.seriesZSeries)zpandas.core.sparse.seriesZSparseSeries)zpandas._libs.sparse
BlockIndex)pandas._libs.tslib	Timestamp)zpandas._libs.tslibs.periodPeriod)zpandas._libs.tslibs.nattype__nat_unpickle)zpandas.core.arrays.sparser    )pandas.core.indexes.base
_new_Index)r*   r   )zpandas.core.indexes.range
RangeIndex)zpandas.core.indexes.multi
MultiIndex)pandas.core.indexes.datetimes_new_DatetimeIndex)r.   DatetimeIndex)zpandas.core.indexes.periodPeriodIndex)r   Categorical)zpandas.core.indexes.timedeltasTimedeltaIndex)zpandas.core.dtypes.dtypesSparseDtype))zpandas.core.sparse.arrayr    )pandas.core.baseFrozenNDArray)r"   r6   )r5   r#   )r$   Z
TimeSeries)zpandas.sparse.seriesZSparseTimeSeries)zpandas._sparser%   )pandas.tslibr'   )zpandas._periodr(   )zpandas._libs.periodr(   )r7   r)   )r&   r)   )zpandas.sparse.arrayr    )pandas.indexes.baser+   )r8   r   )pandas.indexes.numeric
Int64Index)zpandas.indexes.ranger,   )zpandas.indexes.multir-   )pandas.tseries.indexr/   )r;   r0   )zpandas.tseries.periodr1   )zpandas.core.categoricalr2   )zpandas.tseries.tdir3   )r9   Float64Index)pandas.core.indexes.numericr:   )r=   ZUInt64Index)r=   r<   )zpandas.core.arrays.sparse.dtyper4   c                      s   e Zd Z fddZ  ZS )	Unpicklerc                   s&   ||f}t ||\}}t ||S N)_class_locations_mapgetsuper
find_class)r   modulenamekey	__class__r   r   rC      s    zUnpickler.find_class)__name__
__module____qualname__rC   __classcell__r   r   rG   r   r>      s   r>   None)returnc                 C  s   | j  }| j d }t|tr*t|}nt|trX|sXtjg dd}||||j	}n^t|t
r|stjg dd}||||j	}n0|tu r|s||dg d}n|j|g|R  }|| j d< d S )Nr   zM8[ns])dtypezm8[ns]r   F)r   r   r   r   r   r   r   nparrayrO   r	   r
   )r   r   r   objZarrr   r   r   load_newobj   s    


rS   c                 C  sZ   | j  }| j  }| j  }t|tr4t|}n|j|g|R i |}| | d S r?   )r   r   r   r   r   r   append)r   kwargsr   r   rR   r   r   r   load_newobj_ex   s    



rV   Fz
str | Nonebool)encoding
is_verbosec              	   C  sV   z8|  d |dur"t| |d}nt| }||_| W S  ttfyP    Y n0 dS )z
    Load a pickle, with a provided encoding,

    Parameters
    ----------
    fh : a filelike object
    encoding : an optional encoding
    is_verbose : show exception output
    r   N)rX   )seekr>   rY   load
ValueErrorr   )fhrX   rY   Zupr   r   r   r[      s    


r[   TASCIIstrictfix_importsrX   errorsbytesr   )bytes_objectra   rX   rb   c                C  s   t | }t||||d S )z%
    Analogous to pickle._loads.
    r`   )ioBytesIOr>   r[   )rd   ra   rX   rb   fdr   r   r   loads   s    

rh   zGenerator[None, None, None]c               
   c  s<   t j} z"tt dt dV  W tt d|  ntt d|  0 dS )z8
    Temporarily patch pickle to use our unpickler.
    rh   N)pklrh   setattr)Z
orig_loadsr   r   r   patch_pickle   s
    rk   )NF)*__doc__
__future__r   
contextlibcopyre   pickleri   typingr   r!   rP   Zpandas._libs.arraysr   Zpandas._libs.tslibsr   Zpandasr   Zpandas.core.arraysr   r   r	   Zpandas.core.internalsr
   collections.abcr   r   r@   
_Unpicklerr>   dispatchREDUCErS   NEWOBJrV   	NEWOBJ_EXAttributeErrorKeyErrorr[   rh   contextmanagerrk   r   r   r   r   <module>   sz   $W