Sharing Full Python Code for Reproducible Issue

happy to provide the full python code I am using so that the issue can be reproduced also