mirror of
https://github.com/wassname/ray.git
synced 2026-06-28 03:34:48 +08:00
Sven
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
58 lines
2.3 KiB
Python
Executable File
58 lines
2.3 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 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
|
|
if os.access(os.path.dirname(package_home), os.W_OK):
|
|
subprocess.check_call(["rm", "-rf", package_home])
|
|
subprocess.check_call(["ln", "-s", local_home, package_home])
|
|
else:
|
|
print("You don't have write permission to {}, using sudo:".format(
|
|
package_home))
|
|
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)
|
|
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://ray.readthedocs.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.")
|