From 1c9584076549054a5d1ee1fd6fe95104b229a43c Mon Sep 17 00:00:00 2001
From: Robert Nishihara <robertnishihara@gmail.com>
Date: Tue, 13 Dec 2016 19:24:27 -0800
Subject: [PATCH] Revert cloudpickle -> pickling change, which no longer seems
 necessary. (#127)

---
 lib/python/ray/worker.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/lib/python/ray/worker.py b/lib/python/ray/worker.py
index 8db23cc6a..ae1b420a3 100644
--- a/lib/python/ray/worker.py
+++ b/lib/python/ray/worker.py
@@ -2,7 +2,6 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import cloudpickle
 import hashlib
 import os
 import sys
@@ -521,7 +520,7 @@ class Worker(object):
       function_to_run_id = random_string()
       key = "FunctionsToRun:{}".format(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.driver_export_counter += 1