Machine Learning in Elixir: Chap8 axon_onnx 0.4 dependency does not compile

In the list of dependencies to install in the chapter Identifying Cats and Dogs Again, the dependency :axon_onnx 0.4 does not compile

Compilation error in file lib/axon_onnx/shared.ex ==

** (CompileError) lib/axon_onnx/shared.ex:58: undefined function transform/2 (there is no such import

I am using LiveBook version 0.16.4 , any suggestion to fix that ?