Ray Crash Test - No Prefect

ray_test_noprefect.py

import ray
import time


class foo:
    def __init__(self):
        pass

    def run(self, i):
        return f"hi {i}"


def test():
    f = ray.remote(num_gpus=0)(foo)
    pool = ray.util.ActorPool([f.remote()])
    print(pool)
    results = pool.map(lambda a, i: a.run.remote(i), [1])
    print(list(results))
    print("done")


def sleep():
    time.sleep(60 * 10)


def ray_test():
    test()
    sleep()


if __name__ == "__main__":
    ray_test()