From 5b7d235052da3de54ce95a8f0d90056d0f9b3b93 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Mon, 9 Oct 2023 16:22:36 -0500
Subject: [PATCH] snapshot...

---
 modules/util/infer_cloud_products.py | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/modules/util/infer_cloud_products.py b/modules/util/infer_cloud_products.py
index 7cbad5af..d34ac002 100644
--- a/modules/util/infer_cloud_products.py
+++ b/modules/util/infer_cloud_products.py
@@ -44,10 +44,15 @@ def infer_cloud_products(clvrx_path, output_dir, full_disk=True, satellite='GOES
         t0 = time.time()
         if full_disk:
             cld_frac = nn_cld_frac.run_inference_full_disk(pname, None)
-            cld_opd = nn_cld_opd.run_inference_full_disk(pname, None)
+            cld_opd, opd = nn_cld_opd.run_inference_full_disk(pname, None)
         else:
             cld_frac = nn_cld_frac.run_inference(pname, None)
-            cld_opd = nn_cld_opd.run_inference(pname, None)
+            cld_opd, opd = nn_cld_opd.run_inference(pname, None)
+
+        clr_cat = cld_frac == 0
+        cld_cat = cld_frac == 4
+        cld_opd[clr_cat] = opd[clr_cat]
+        cld_opd[cld_cat] = opd[cld_cat]
 
         write_cld_prods_file_nc4(clvrx_str_time, out_file, cld_frac, cld_opd, x_rad, y_rad, None, None,
                                 satellite=satellite, domain=domain, has_time=True)
-- 
GitLab