Hi! I’m trying out diabates classification example, but with remote execution the split_traitest_dataset task fails because there is no pandas module I think it is because flyte default image doesn’t contain this module. Is there any prebuilt images that I can use with pandas and other dependencies, or I have to build it myself?

@steep-nest-3156 you should be able to use ImageSpec to declare the custom image with the required dependencies. In that way you don't have to write a Dockerfile

Works, thank you! Or at least I’ve got another error with pulling a container. but I think it is connected with M1 macbook

if you're running it locally, please add

platform=darwin/arm64

to your ImageSpec config

Thank you, this doesn’t work me as well, now on a building step

ERROR: failed to solve: <http://ghcr.io/astral-sh/uv:0.2.37|ghcr.io/astral-sh/uv:0.2.37>: failed to resolve source metadata for <http://ghcr.io/astral-sh/uv:0.2.37|ghcr.io/astral-sh/uv:0.2.37>: no match for platform in manifest: not found

sorry, try with

platform = "linux/arm64"

Also I needed “registry=localhost:30000” It works now, thank you!

oh sorry for the struggles, feel free to contribute to the example by pointing to these missing pieces