rayclient/python/example_client.py
2020-10-15 19:20:34 +00:00

34 lines
561 B
Python

import cloudpickle
import logging
from concurrent import futures
import grpc
import ray
ray.connect("localhost:50051")
@ray.remote
def fact(x):
out = 1
for i in range(1, x+1):
out = i * out
return out
@ray.remote
def fib(x):
if x <= 1:
return 1
return ray.get(fib.remote(x - 1)) + ray.get(fib.remote(x - 2))
def run():
out = fact.remote(5)
out2 = fact.remote(out)
print("Got fact: ", ray.get(out2))
fib_out = fib.remote(6)
print("Fib out:", ray.get(fib_out))
if __name__ == "__main__":
run()