[Serve] Use pickle instead of clouldpickle (#9479)

This commit is contained in:
Simon Mo
2020-07-16 15:44:10 -07:00
committed by GitHub
parent 761b584814
commit 60a838f0f0
4 changed files with 49 additions and 3 deletions
+2 -2
View File
@@ -3,10 +3,10 @@ import copy
from collections import defaultdict, deque
import time
from typing import DefaultDict, List
import pickle
import blist
import ray.cloudpickle as pickle
from ray.exceptions import RayTaskError
import ray
@@ -50,7 +50,7 @@ class Query:
# worker without removing async_future.
clone = copy.copy(self).__dict__
clone.pop("async_future")
return pickle.dumps(clone, protocol=5)
return pickle.dumps(clone)
@staticmethod
def ray_deserialize(value):