See initial discussion at https://github.com/oumi-ai/roumi/pull/20#discussion_r2067692509 It is probably a good idea to clean this up a bit and not require a plugin developer to copy the code for the trait -