From 7d1d308f06857f37b260110f6954e4ab31659319 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Fri, 21 Apr 2023 11:04:54 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/cloud_fraction_fcn_viirs.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/modules/deeplearning/cloud_fraction_fcn_viirs.py b/modules/deeplearning/cloud_fraction_fcn_viirs.py
index bfab3245..cd5bf4b1 100644
--- a/modules/deeplearning/cloud_fraction_fcn_viirs.py
+++ b/modules/deeplearning/cloud_fraction_fcn_viirs.py
@@ -772,7 +772,7 @@ def run_restore_static(directory, ckpt_dir, out_file=None):
     labels, preds, inputs = nn.run_restore(directory, ckpt_dir)
     if out_file is not None:
         np.save(out_file,
-                [np.squeeze(labels), preds.argmax(axis=3), np.squeeze(inputs)])
+                [np.squeeze(labels), preds.argmax(axis=3), inputs])
 
 
 def run_evaluate_static(in_file, out_file, ckpt_dir):
-- 
GitLab