Hi @arturn
Thanks for taking a look at this. It’s been a while indeed but I remember that I had tried output_indexes=[0, 1] but I believe it didn’t work or kept giving me the same output. I will see if I can rerun this and let you know if there was any improvement.