angle wrapping LQR?

[truher]

one of my students has gotten a jump on the summer session by prototyping a 1-DOF arm with LQR control, but they don't know how to do the wrapping. the docs suggest replacing lqr.calculate() but the example code doesn't use calculate() at all. could you suggest an approach to wrapping that fits the current example LQR arm example code?

https://github.com/wpilibsuite/frc-docs/blob/f2b037b939284fd4bb3e21cc6b3cb70de6236e64/source/docs/software/advanced-controls/state-space/state-space-flywheel-walkthrough.rst?plain=1#L258