wehub-resource-sync
31 lines
676 B
Python
31 lines
676 B
Python
from pickle import (
|
|
_Pickler,
|
|
_Unpickler as Unpickler,
|
|
_loads as loads,
|
|
_load as load,
|
|
PickleError,
|
|
PicklingError,
|
|
UnpicklingError,
|
|
HIGHEST_PROTOCOL,
|
|
)
|
|
|
|
__all__ = [
|
|
"PickleError",
|
|
"PicklingError",
|
|
"UnpicklingError",
|
|
"Pickler",
|
|
"Unpickler",
|
|
"load",
|
|
"loads",
|
|
"HIGHEST_PROTOCOL",
|
|
]
|
|
|
|
|
|
class Pickler(_Pickler):
|
|
def __init__(self, file, protocol=None, *, fix_imports=True, buffer_callback=None):
|
|
super().__init__(
|
|
file, protocol, fix_imports=fix_imports, buffer_callback=buffer_callback
|
|
)
|
|
# avoid being overrided by cloudpickle
|
|
self.dispatch = _Pickler.dispatch.copy()
|