Write computation graph to file

lib/python/ray/__init__.py

@@ -8,5 +8,6 @@ PYTHON_MODE = 3
 
 import libraylib as lib
 import serialization
-from worker import scheduler_info, task_info, register_module, connect, disconnect, get, put, remote
+from worker import scheduler_info, dump_computation_graph, task_info, register_module, connect, disconnect, get, put, remote
 from libraylib import ObjRef
+import internal

lib/python/ray/worker.py

@@ -143,6 +143,9 @@ def print_task_info(task_data, mode):
 def scheduler_info(worker=global_worker):
   return ray.lib.scheduler_info(worker.handle);
 
+def dump_computation_graph(file_name, worker=global_worker):
+  ray.lib.dump_computation_graph(worker.handle, file_name)
+
 def task_info(worker=global_worker):
   """Tell the scheduler to return task information. Currently includes a list of all failed tasks since the start of the cluster."""
   return ray.lib.task_info(worker.handle);