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
This commit is contained in:
Wapaul1
2016-08-25 15:26:22 -07:00
committed by Robert Nishihara
parent 43452b9ab3
commit 420bcc0477
3 changed files with 21 additions and 2 deletions
+12
View File
@@ -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)