1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
from typing import Optional
import cloudpickle
from .util import JSONInput, JSONOutput
def pickle_dumps(data: JSONInput, protocol: Optional[int] = None) -> bytes:
"""Serialize a Python object with pickle.
data: The object to serialize.
protocol (int): Protocol to use. -1 for highest.
RETURNS (bytes): The serialized object.
"""
return cloudpickle.dumps(data, protocol=protocol)
def pickle_loads(data: bytes) -> JSONOutput:
"""Deserialize bytes with pickle.
data (bytes): The data to deserialize.
RETURNS: The deserialized Python object.
"""
return cloudpickle.loads(data)
|
