o ki@sddlZddlmZmZmZmZmZmZmZm Z m Z m Z ddl m Z ddlmZddlmZmZmZmZGdddeZGd d d eZdd d ZdS)N) _compat_pickle_extension_registry _getattribute_PicklerEXT1EXT2EXT4GLOBAL PicklingError STACK_GLOBAL)pack) FunctionType)ImporterObjMismatchErrorObjNotFoundError sys_importercs(eZdZfddZfddZZS)_PyTorchLegacyPicklercstj|i|d|_dSN)super__init___persistent_id)selfargskwargs __class__h/var/www/addictedbytheproject.nl/epg/venv/lib/python3.10/site-packages/torch/package/_package_pickler.pyrs z_PyTorchLegacyPickler.__init__cs |jdur t|S||Sr)rr persistent_id)robjrrrrs   z#_PyTorchLegacyPickler.persistent_id)__name__ __module__ __qualname__rr __classcell__rrrrrs rcs0eZdZdZdeffdd ZdddZZS) PackagePicklerzPackage-aware pickler. This behaves the same as a normal pickler, except it uses an `Importer` to find objects and modules to save. importercs4||_tj|i|tj|_tj|jt<dSr) r&rrrdispatchcopyr% save_globalr )rr&rrrrrr&s zPackagePickler.__init__Nc Cs|j}|j}z |j||\}}Wnttfy,}z td|dt||d}~ww|j|}t j dkr[| d}d|vrHtd|t |dkrQ|} nt ||dd} nt ||\} } |jd krt||f} | r| d kr{td | | d kr|ttd | dS| dkr|ttd| dS|ttd| dS|dd } | |ur| }|jdkr|||||tnz| |ur|t| | fnm|jdkr|tt|ddt|ddnU|jr tj} tj }||f| vr| ||f\}}n ||vr ||}z|tt|ddt|ddWnt!y>}ztd|d|d|jd|d}~ww|"|dS)Nz Can't pickle z: ).zzCan't pickle local object rrz&expected positive extension code, got zs0  k