Compilation error in file lib/axon_onnx/shared.ex

[TheEndIsNear]

I am getting the following error when attempting to install axon_onnx in livebook.

== 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) (nx 0.7.1) lib/nx/defn/compiler.ex:832: Nx.Defn.Compiler.compile_error!/3 (nx 0.7.1) lib/nx/defn/compiler.ex:429: Nx.Defn.Compiler.normalize/2 (elixir 1.16.2) lib/enum.ex:1826: Enum."-map_reduce/3-lists^mapfoldl/2-0-"/3 (elixir 1.16.2) lib/enum.ex:1826: Enum."-map_reduce/3-lists^mapfoldl/2-0-"/3 (nx 0.7.1) lib/nx/defn/compiler.ex:422: Nx.Defn.Compiler.normalize/2 (elixir 1.16.2) lib/enum.ex:1826: Enum."-map_reduce/3-lists^mapfoldl/2-0-"/3 (elixir 1.16.2) lib/enum.ex:1826: Enum."-map_reduce/3-lists^mapfoldl/2-0-"/3 (nx 0.7.1) lib/nx/defn/compiler.ex:422: Nx.Defn.Compiler.normalize/2

The command that I am running in the setup.

Mix.install([ {:axon_onnx, "~> 0.4"}, {:axon, "~> 0.5"}, {:nx, "~> 0.5"}, {:exla, "~> 0.5"}, {:stb_image, "~> 0.6"}, {:kino, "~> 0.8"}, ])

I am running elixir 1.16.2 with OTP/26 with Livebook 0.12.1 on Ubuntu 20.04 in Windows Subsystem for Linux.

[josevalim]

You can use main for now.

[TheEndIsNear]

That fixed it thanks Jose!

[khari998]

Getting the same error

[Eiji7]

2 months without any update? Today someone asked about this issue on #Elixir Chat too.

[khari998]

2 months without any update? Today someone asked about this issue on #Elixir Chat too.

I switched to using Ortex instead. Even on the latest main branch from the repo I was having other issues.