- prevents print statements from occuring for every MPI rank

Hi @gubaidulinvadim, I found that this change also requires importing the sys module when I tried to run it. I had to add 4c54d8ac for it to work. Does it work for you without adding that import?

Hi @TeresiaOlsson, I must admit that I might not have tested this commit yet, as it is still in the develop branch. I haven't recreated a new container for my scripts with this version yet, so adding "sys" import is a good addition!