Badger 2040; partial_update() inside a loop


I’m hoping I’ve done something daft but I’m struggling to get partial_update working.
I opened an issue on the github here: Badger 2040; Badger OS; partial_update followed by sleep seems to stop execution · Issue #348 · pimoroni/pimoroni-pico · GitHub
but wondered if anyone else had struggled to get this working inside a loop - first time it seems to work fine but then everything stops for me.

Many thanks,