[BUG] Develop branch hung when using replica > 2

[zye1996]

Describe the bug
I installed latest develop branch and run basic generation with StepResources(replica=8) and the generation hung. It does not happen to tags 88615c7

To Reproduce
Code to reproduce

Expected behaviour
A clear and concise description of what you expected to happen.

Screenshots
If applicable, add screenshots to help explain your problem.

Desktop (please complete the following information):

• Package version:
• Python version:

Additional context
Add any other context about the problem here.

[gabrielmbmb]

Hi @zye1996, thanks for reporting the issue. I'll check!

[gabrielmbmb]

Hi again @zye1996! I confirmed that it gets hung. Just to confirm, at which moment of the execution of the pipeline do you see it gets hung? do you see that some replicas of the step finish?

[gabrielmbmb]

we just merged a PR to develop that fixed one issue that made the pipeline hung just at the end of the execution because not all steps replicas were being unloaded. If you can try and confirm that your issue has been fixed would be awesome!