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.