Revert cloudpickle -> pickling change, which no longer seems necessary. (#127)

This commit is contained in:
Robert Nishihara 2016-12-13 19:24:27 -08:00
parent b0ba54e4c0
commit 1c95840765

View file

@ -2,7 +2,6 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
import cloudpickle
import hashlib import hashlib
import os import os
import sys import sys
@ -521,7 +520,7 @@ class Worker(object):
function_to_run_id = random_string() function_to_run_id = random_string()
key = "FunctionsToRun:{}".format(function_to_run_id) key = "FunctionsToRun:{}".format(function_to_run_id)
self.redis_client.hmset(key, {"function_id": function_to_run_id, self.redis_client.hmset(key, {"function_id": function_to_run_id,
"function": cloudpickle.dumps(function)}) "function": pickling.dumps(function)})
self.redis_client.rpush("Exports", key) self.redis_client.rpush("Exports", key)
self.driver_export_counter += 1 self.driver_export_counter += 1