yes that's what I meant in the comment https://github.com/PyTorchLightning/pytorch-lightning/pull/10622#discussion_r752622231 over here, as we can disentangle and simplify this.   The code was already like this before we moved it to SLURMEnvironment. 