Hey @checcopo!
We had a similar question here: 'NoneType' object is not iterable error when working with mid-circuit measurements or qml.ctrl
I think you’d be able to accomplish what you want to do by following my response to that question. Let me know!