I'm getting some issues with Flyte 2.0 image builder Source code:

ENV = flyte.TaskEnvironment(
    name="my_env",
    image=flyte.Image.from_uv_script(
            __file__,
            name="beacon2_pdf", # 
            registry="<http://099604434934.dkr.ecr.us-east-2.amazonaws.com|099604434934.dkr.ecr.us-east-2.amazonaws.com>"
        )
)

@ENV.task()
def split_pdf_pages(file_path: str, dpi: int, output_dir: str):
    print(f"Splitting {file_path} into pages")
    document = fitz.open(file_path)
    if not os.path.exists(output_dir):
        os.makedirs(output_dir)
    for page_num in range(len(document)):
        page = document[page_num]
        pix = page.get_pixmap(dpi=dpi)
        output_filename = f"p{page_num + 1}.png"
        output_path = os.path.join(output_dir, output_filename)
        print(output_path)
        #pix.save(output_path)


if __name__ == "__main__":
    flyte.init_from_config("config.yaml")
    run = flyte.with_runcontext(mode="remote").run(split_pdf_pages,
                file_path="/Users/adong/beacon/playground/split_test.pdf",
                dpi=300,
                output_dir="/Users/adong/beacon/playground/output/",
    )
    print(run)

Error is attached

Hey @dry-shampoo-96961! I am moving this to the other slack connect channel we have with you. I tagged you there!