You should use pm.set_data in this case. There is an applied example here that seems quite similar to the code you’ve shared.
You should use pm.set_data in this case. There is an applied example here that seems quite similar to the code you’ve shared.