Change logfile names and also allow plasma store socket to be passed in. (#2862)

python/ray/workers/default_worker.py

@@ -9,6 +9,7 @@ import traceback
 import ray
 import ray.actor
 import ray.ray_constants as ray_constants
+import ray.tempfile_services as tempfile_services
 
 parser = argparse.ArgumentParser(
     description=("Parse addresses for the worker "
@@ -53,6 +54,12 @@ parser.add_argument(
     type=str,
     default=ray_constants.LOGGER_FORMAT,
     help=ray_constants.LOGGER_FORMAT_HELP)
+parser.add_argument(
+    "--temp-dir",
+    required=False,
+    type=str,
+    default=None,
+    help="Specify the path of the temporary directory use by Ray process.")
 
 if __name__ == "__main__":
     args = parser.parse_args()
@@ -70,6 +77,9 @@ if __name__ == "__main__":
         level=logging.getLevelName(args.logging_level.upper()),
         format=args.logging_format)
 
+    # Override the temporary directory.
+    tempfile_services.set_temp_root(args.temp_dir)
+
     ray.worker.connect(
         info, mode=ray.WORKER_MODE, use_raylet=(args.raylet_name is not None))