Merge pull request #64 from oceanmodeling/bugfix/nointernet_geods

Cache geodataset path in the section that runs on head node
oceanmodeling · Aug 15, 2024 · d42cb88 · d42cb88
2 parents d898f50 + d66cb72
commit d42cb88
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/stormworkflow/prep/hurricane_data.py b/stormworkflow/prep/hurricane_data.py
@@ -17,6 +17,7 @@
 
 import pandas as pd
 import geopandas as gpd
+import geodatasets
 from searvey.coops import COOPS_TidalDatum
 from searvey.coops import COOPS_TimeZone
 from searvey.coops import COOPS_Units
@@ -147,6 +148,9 @@ def main(args):
     if hr_before_landfall < 0:
         hr_before_landfall = 48
 
+    # Caching for next steps!
+    geodatasets.get_path('naturalearth land')
+
     ne_low = gpd.read_file(NE_LOW_ADMIN)
     shp_US = ne_low[ne_low.NAME_EN.isin(['United States of America', 'Puerto Rico'])].unary_union