Fix axes flattening for single subplot case#3889

Sgt-Hashtag wants to merge 2 commits intoisaac-sim:mainisaac-sim/IsaacLab:mainfrom

Sgt-Hashtag:mainSgt-Hashtag/IsaacLab:mainCopy head branch name to clipboard