Merge pull request #10 from ihmeuw/feature/precipitation-days

Add precipitation days

Author: collijk

src/climate_data/data.py

@@ -163,6 +163,8 @@ def save_daily_results(
         encoding_kwargs: dict[str, Any],
     ) -> None:
         path = self.daily_results_path(scenario, variable, year)
+        if path.exists():
+            path.unlink()
         mkdir(path.parent, exist_ok=True, parents=True)
         touch(path, exist_ok=True)
 

src/climate_data/generate/scenario_annual.py

@@ -152,6 +152,10 @@
         transform_funcs=[utils.annual_sum],
         encoding_scale=10,
     ),
+    "precipitation_days": utils.Transform(
+        source_variables=["total_precipitation"],
+        transform_funcs=[utils.count_threshold(0.1), utils.annual_sum],
+    ),
 }