Hi, I am trying to setup a flake for my rust program but I am having some issues...

I am using the Burn framework for neural networks and am importing an ONNX pre-trained models in a build.rs script. The problem is that when I run nix build (I am using buildRustPackage) it fails due to not being able to find the .onnx files (No such file or directory). Do I somehow need to tell Nix about the .onnx files for the build to work?

(I am able to compile just fine in a dev shell)

(I hope this is the right room to ask in)