compress samples on disk in predict.py

henryaddison · Aug 3, 2024 · f6a7cd0 · f6a7cd0
1 parent 5b8268e
commit f6a7cd0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/bin/predict.py b/bin/predict.py
@@ -276,6 +276,8 @@ def main(
         output_filepath = output_dirpath / f"predictions-{shortuuid.uuid()}.nc"
 
         logger.info(f"Saving samples to {output_filepath}...")
+        for varname in xr_samples.data_vars:
+            xr_samples[varname].encoding.update(zlib=True, complevel=5)
         xr_samples.to_netcdf(output_filepath)