[helios]: Minor mod to plot generation script
[helios]: Minor mod to plot generation script

transcendent-ai-labs/PlasmaML,transcendent-ai-labs/PlasmaML,mandar2812/PlasmaML,mandar2812/PlasmaML,transcendent-ai-labs/PlasmaML