mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 01:27:43 +08:00
Fix setup (#5400)
This commit is contained in:
@@ -13,11 +13,11 @@ import subprocess
|
||||
import ray
|
||||
|
||||
|
||||
def do_link(package, force=False):
|
||||
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__, "../{}".format(package)))
|
||||
os.path.join(__file__, local_path + "../{}".format(package)))
|
||||
assert os.path.isdir(package_home), package_home
|
||||
assert os.path.isdir(local_home), local_home
|
||||
if not force and not click.confirm(
|
||||
@@ -43,7 +43,7 @@ if __name__ == "__main__":
|
||||
"--yes", action="store_true", help="Don't ask for confirmation.")
|
||||
args = parser.parse_args()
|
||||
|
||||
do_link("rllib", force=args.yes)
|
||||
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)
|
||||
|
||||
Reference in New Issue
Block a user