17 lines
344 B
Python
17 lines
344 B
Python
# from rpc import ray_client_pb2
|
|
# from rpc import ray_client_pb2_grpc
|
|
from ray_worker import Worker
|
|
import sys
|
|
from ray import ray
|
|
|
|
|
|
def main():
|
|
worker = Worker(sys.argv[1])
|
|
ray.connect(sys.argv[1], stub=worker.server)
|
|
worker.begin()
|
|
print("Shutting down...")
|
|
worker.channel.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|