@mcabbott is right, I had a typo there, where I have not fixed the problem of how data should be represented. I have fixed the example and it should be right now. @Jcverni Can you check it again?
For completness, I tested the script with julia 1.9.0 and the package versions are
Status `/private/tmp/Project.toml`
[587475ba] Flux v0.13.16
[eb30cadb] MLDatasets v0.7.9