mirror of
https://github.com/wassname/ray.git
synced 2026-06-27 22:38:16 +08:00
Sven Mika
72 lines
2.7 KiB
Python
Executable File
72 lines
2.7 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""This script allows you to develop RLlib without needing to compile Ray."""
|
|
|
|
import argparse
|
|
import click
|
|
import os
|
|
import shutil
|
|
import subprocess
|
|
|
|
import ray
|
|
|
|
|
|
def do_link(package, force=False, local_path=""):
|
|
package_home = os.path.abspath(
|
|
os.path.join(ray.__file__, "../{}".format(package)))
|
|
local_home = os.path.abspath(
|
|
os.path.join(__file__, local_path + "../{}".format(package)))
|
|
if not os.path.isdir(package_home):
|
|
print("{} does not exist. Continuing to link.".format(package_home))
|
|
assert os.path.isdir(local_home), local_home
|
|
if not force and not click.confirm(
|
|
"This will replace:\n {}\nwith a symlink to:\n {}".format(
|
|
package_home, local_home),
|
|
default=True):
|
|
return
|
|
# Windows: Create directory junction.
|
|
if os.name == "nt":
|
|
try:
|
|
shutil.rmtree(package_home)
|
|
except FileNotFoundError:
|
|
pass
|
|
except OSError:
|
|
os.remove(package_home)
|
|
subprocess.check_call(
|
|
["mklink", "/J", package_home, local_home], shell=True)
|
|
# Posix: Use `ln -s` to create softlink.
|
|
else:
|
|
sudo = []
|
|
if not os.access(os.path.dirname(package_home), os.W_OK):
|
|
print("You don't have write permission to {}, using sudo:".format(
|
|
package_home))
|
|
sudo = ["sudo"]
|
|
subprocess.check_call(sudo + ["rm", "-rf", package_home])
|
|
subprocess.check_call(sudo + ["ln", "-s", local_home, package_home])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(
|
|
formatter_class=argparse.RawDescriptionHelpFormatter,
|
|
description="Setup dev.")
|
|
parser.add_argument(
|
|
"--yes", action="store_true", help="Don't ask for confirmation.")
|
|
args = parser.parse_args()
|
|
|
|
do_link("rllib", force=args.yes, local_path="../../")
|
|
do_link("tune", force=args.yes)
|
|
do_link("autoscaler", force=args.yes)
|
|
do_link("scripts", force=args.yes)
|
|
do_link("internal", force=args.yes)
|
|
do_link("tests", force=args.yes)
|
|
do_link("experimental", force=args.yes)
|
|
do_link("util", force=args.yes)
|
|
do_link("dashboard", force=args.yes)
|
|
print("Created links.\n\nIf you run into issues initializing Ray, please "
|
|
"ensure that your local repo and the installed Ray are in sync "
|
|
"(pip install -U the latest wheels at "
|
|
"https://docs.ray.io/en/latest/installation.html, "
|
|
"and ensure you are up-to-date on the master branch on git).\n\n"
|
|
"Note that you may need to delete the package symlinks when pip "
|
|
"installing new Ray versions to prevent pip from overwriting files "
|
|
"in your git repo.")
|