Yeah, it’s a bit weird, but your guess might be right. The issue might indeed come from a difference in the data loading or processing.
Could you print the shape of action in both envs so that we could maybe try debugging the actual issue to narrow down the difference?