Skip to content

Commit

Permalink
Remove stage_ic job dependencies from setup
Browse files Browse the repository at this point in the history
  • Loading branch information
KateFriedman-NOAA committed May 23, 2024
1 parent 9a39787 commit cd362fd
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 96 deletions.
45 changes: 0 additions & 45 deletions workflow/rocoto/gefs_tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,55 +12,10 @@ def stage_ic(self):

cpl_ic = self._configs['stage_ic']

deps = []

# Atm ICs
if self.app_config.do_atm:
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_ATMIC']}/@Y@m@d@H/mem000/atmos"
for file in ['gfs_ctrl.nc'] + \
[f'{datatype}_data.tile{tile}.nc'
for datatype in ['gfs', 'sfc']
for tile in range(1, self.n_tiles + 1)]:
data = f"{prefix}/{file}"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

# Ocean ICs
if self.app_config.do_ocean:
ocn_res = f"{self._base.get('OCNRES', '025'):03d}"
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_OCNIC']}/@Y@m@d@H/mem000/ocean"
data = f"{prefix}/@Y@m@d.@H0000.MOM.res.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))
if ocn_res in ['025']:
# 0.25 degree ocean model also has these additional restarts
for res in [f'res_{res_index}' for res_index in range(1, 4)]:
data = f"{prefix}/@Y@m@d.@H0000.MOM.{res}.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

# Ice ICs
if self.app_config.do_ice:
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_ICEIC']}/@Y@m@d@H/mem000/ice"
data = f"{prefix}/@Y@m@d.@H0000.cice_model.res.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

# Wave ICs
if self.app_config.do_wave:
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_WAVIC']}/@Y@m@d@H/mem000/wave"
for wave_grid in self._configs['waveinit']['waveGRD'].split():
data = f"{prefix}/@Y@m@d.@H0000.restart.{wave_grid}"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

dependencies = rocoto.create_dependency(dep_condition='and', dep=deps)

resources = self.get_resource('stage_ic')
task_name = f'stage_ic'
task_dict = {'task_name': task_name,
'resources': resources,
'dependency': dependencies,
'envars': self.envars,
'cycledef': 'gefs',
'command': f'{self.HOMEgfs}/jobs/rocoto/stage_ic.sh',
Expand Down
51 changes: 0 additions & 51 deletions workflow/rocoto/gfs_tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,61 +20,10 @@ def stage_ic(self):

cpl_ic = self._configs['stage_ic']

deps = []

# Atm ICs
if self.app_config.do_atm:
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_ATMIC']}/@Y@m@d@H/atmos"
for file in ['gfs_ctrl.nc'] + \
[f'{datatype}_data.tile{tile}.nc'
for datatype in ['gfs', 'sfc']
for tile in range(1, self.n_tiles + 1)]:
data = f"{prefix}/{file}"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))
else: # data-atmosphere
# TODO - need more information about how these forcings are stored
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_DATM']}/@Y@m@d@H"
data = f"{prefix}/gefs.@Y@m.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

# Ocean ICs
if self.app_config.do_ocean:
ocn_res = f"{self._base.get('OCNRES', '025'):03d}"
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_OCNIC']}/@Y@m@d@H/ocean"
data = f"{prefix}/@Y@m@d.@H0000.MOM.res.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))
if ocn_res in ['025']:
# 0.25 degree ocean model also has these additional restarts
for res in [f'res_{res_index}' for res_index in range(1, 4)]:
data = f"{prefix}/@Y@m@d.@H0000.MOM.{res}.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

# Ice ICs
if self.app_config.do_ice:
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_ICEIC']}/@Y@m@d@H/ice"
data = f"{prefix}/@Y@m@d.@H0000.cice_model.res.nc"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

# Wave ICs
if self.app_config.do_wave:
prefix = f"{cpl_ic['BASE_CPLIC']}/{cpl_ic['CPL_WAVIC']}/@Y@m@d@H/wave"
for wave_grid in self._configs['waveinit']['waveGRD'].split():
data = f"{prefix}/@Y@m@d.@H0000.restart.{wave_grid}"
dep_dict = {'type': 'data', 'data': data}
deps.append(rocoto.add_dependency(dep_dict))

dependencies = rocoto.create_dependency(dep_condition='and', dep=deps)

resources = self.get_resource('stage_ic')
task_name = f'{self.cdump}stage_ic'
task_dict = {'task_name': task_name,
'resources': resources,
'dependency': dependencies,
'envars': self.envars,
'cycledef': self.cdump,
'command': f'{self.HOMEgfs}/jobs/rocoto/stage_ic.sh',
Expand Down

0 comments on commit cd362fd

Please sign in to comment.