wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-27 16:31:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update example shuffle script (#14021)

Browse Source
This commit is contained in:
Eric Liang
2021-02-09 20:47:41 -08:00
committed by GitHub
parent 79c7c181f3
commit 7f342eb371
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/experimental/shuffle.py
+4 -2
View File
@@ -169,8 +169,10 @@ def main():
parser.add_argument("--partition-size", type=float, default=200e6)
args = parser.parse_args()
ray.init(
address=args.ray_address, object_store_memory=args.object_store_memory)
if args.ray_address:
ray.init(address=args.ray_address)
else:
ray.init(object_store_memory=args.object_store_memory)
partition_size = int(args.partition_size)
num_partitions = args.num_partitions