modal byod test

import sys

import modal
from modal import Image

ray_image = Image.from_registry("anyscale/ray:2.8.0-py39-cu118")
stub = modal.Stub("example-hello-world")


@stub.function(image=ray_image)
def f(i):
    if i % 2 == 0:
        print("hello", i)
    else:
        print("world", i, file=sys.stderr)

    return i * i

@stub.local_entrypoint()
def main():
    # Call the function locally.
    print(f.local(1000))

    # Call the function remotely.
    print(f.remote(1000))

    # Parallel map.
    total = 0
    for ret in f.map(range(20)):
        total += ret

    print(total)