diff --git a/ush/soca/run_jjobs.py b/ush/soca/run_jjobs.py index 4ccc862f7..83fe4602e 100755 --- a/ush/soca/run_jjobs.py +++ b/ush/soca/run_jjobs.py @@ -150,12 +150,16 @@ def _conda_envs(self, jjob): """ Write a section that will load the machine dependent modules """ - if self.machine != "container": + if self.machine in ["orion", "hercules"]: if jjob in ENVS: # set +/-u is a workaround for an apparent conda bug self.f.write(f"set +u \n") self.f.write(f"conda activate {ENVS[jjob]} \n") self.f.write(f"set -u \n") + elif self.machine == "hera": + if jjob in ENVS: + self.f.write(f"module unload GDAS \n") + self.f.write(f"module load {ENVS[jjob].upper()/{self.machine}} \n") def precom(self, com, tmpl): cmd = f"RUN={self.RUN} YMD={self.gPDY} HH={self.gcyc} declare_from_tmpl -xr {com}:{tmpl}"