From d0165f67c0a9896d0ec9fac5d33291b989623c20 Mon Sep 17 00:00:00 2001
From: tomrink <rink@ssec.wisc.edu>
Date: Tue, 22 Mar 2022 17:17:08 -0500
Subject: [PATCH] snapshot...

---
 modules/deeplearning/icing_fcn.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 3d31fad0..05e2284d 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -400,6 +400,8 @@ class IcingIntensityFCN:
         nda = np.zeros([1])
         nda[0] = self.flight_level
         nda = tf.one_hot(nda, 5).numpy()
+        nda = np.expand_dims(nda, axis=0)
+        nda = np.expand_dims(nda, axis=0)
 
         return data, nda
 
-- 
GitLab