Merge pull request #61 from oceanmodeling/bugfix/hotstart

Add hotstart ramp fix
oceanmodeling · Aug 9, 2024 · a3141b8 · a3141b8
2 parents 8e2e622 + 14464d8
commit a3141b8
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -35,10 +35,10 @@ dependencies = [
     "cmocean",
     "dask",
     "dask-jobqueue",
-    "ensembleperturbation>=1.1.2",
+    "ensembleperturbation==1.2.10",
     "fiona",
     "geoalchemy2",
-    "geopandas>=0.14",
+    "geopandas",
     "matplotlib",
     "mpi4py",
     "netCDF4",
@@ -53,7 +53,7 @@ dependencies = [
     "pytz",
     "pyyaml",
     "shapely>=2",
-    "stormevents>=2.2.4",
+    "stormevents==2.2.5",
     "rasterio",
     "requests",
     "rtree",

diff --git a/stormworkflow/prep/setup_ensemble.py b/stormworkflow/prep/setup_ensemble.py
@@ -107,6 +107,17 @@ def _fix_nwm_issues(ensemble_dir, hires_shapefile):
             _relocate_source_sink(pth, hires_shapefile)
 
 
+def _fix_hotstart_issue(ensemble_dir):
+    hotstart_dirs = ensemble_dir.glob('runs/*')
+    for pth in hotstart_dirs:
+        nm_list = f90nml.read(pth / 'param.nml')
+        nm_list['opt']['dramp'] = 0.0
+        nm_list['opt']['drampbc'] = 0.0
+        nm_list['opt']['dramp_ss'] = 0.0
+        nm_list['opt']['drampwind'] = 0.0
+        nm_list.write(pth / 'param.nml', force=True)
+
+
 def main(args):
 
     track_path = args.track_file
@@ -267,6 +278,7 @@ def main(args):
         }
     )
 
+    _fix_hotstart_issue(workdir)
     if with_hydrology:
         _fix_nwm_issues(workdir, hires_reg)
     if use_wwm: