I think this is something to do with union serverless pulling in a newer version of the Flytekit sdk than it should. (beta) I should probably pin the version expected in ImageSpec for the tutorial. really should have had pinned versions of everything, but hand't run into that before for Flytekit, I just re-ran everything and this should work more reliably for everyone!

image = ImageSpec(
    packages=[
        "union==0.1.85",
        "flytekit==1.13.8",
        "transformers==4.39.1",
        "datasets==2.18.0",
        "matplotlib==3.8.3",
        "torch==2.0.1",
        "accelerate==0.27.2",
        "scikit-learn==1.5.1"
    ],
)