IIUC self.states will return you the variables containing the states.
You should be able to use var.assign(x) to update them.
But the code may be clearer if you use stateful=False and return_states=True. Then you have direct control of the states, you don’t have to reach into the RNN to edit the.