Remote function returning non-serializable type no longer shuts worker down (#384)

* Moved put_objects in main_loop to inside of try block

* Added test for failed serialization

* Fixed naming

* Minor

test/failure_test.py

@@ -34,6 +34,18 @@ class FailureTest(unittest.TestCase):
 
     ray.worker.cleanup()
 
+  def testUnknownSerialization(self):
+    reload(test_functions)
+    ray.init(start_ray_local=True, num_workers=1, driver_mode=ray.SILENT_MODE)
+
+    test_functions.test_unknown_type.remote()
+    time.sleep(0.2)
+    task_info = ray.task_info()
+    self.assertEqual(len(task_info["failed_tasks"]), 1)
+    self.assertEqual(len(task_info["running_tasks"]), 0)
+
+    ray.worker.cleanup()
+
 class TaskStatusTest(unittest.TestCase):
   def testFailedTask(self):
     reload(test_functions)

test/test_functions.py

@@ -116,3 +116,11 @@ def test_return1():
 @ray.remote([], [int, float])
 def test_return2():
   return 2.0, 3.0
+
+class TestClass(object):
+  def __init__(self):
+    self.a = 5
+
+@ray.remote([], [TestClass])
+def test_unknown_type():
+  return TestClass()