From b82d8ee00471c07b7dcb8db984024f96d42e5f73 Mon Sep 17 00:00:00 2001 From: zmoon Date: Mon, 9 Sep 2024 15:18:06 -0500 Subject: [PATCH] Alternative approach since otherwise it complains about the vars not being defined later --- src/fire_inputgen.py | 21 +++++---------------- 1 file changed, 5 insertions(+), 16 deletions(-) diff --git a/src/fire_inputgen.py b/src/fire_inputgen.py index fdb8bc3..dc900e5 100755 --- a/src/fire_inputgen.py +++ b/src/fire_inputgen.py @@ -645,6 +645,7 @@ def main_driver(initial_hour, forecast_hour, f_input, f_output, lat_lim, lon_lim lw[lw == 0] = np.nan # fire frame + INPUTFRAME = LATFRAME = LONFRAME = None for j in np.arange(1, num + 1, 1): total = total + 1 index = np.argwhere(lw == j) @@ -680,22 +681,10 @@ def main_driver(initial_hour, forecast_hour, f_input, f_output, lat_lim, lon_lim loc[0] - fsize : loc[0] + fsize + 1, loc[1] - fsize : loc[1] + fsize + 1 ] - if "INPUTFRAME" in locals(): - INPUTFRAME = np.append( - INPUTFRAME, - np.expand_dims(X_fire, axis=0), - axis=0, - ) # noqa: F821 - LATFRAME = np.append( - LATFRAME, - np.expand_dims(X_lat, axis=0), - axis=0, - ) # noqa: F821 - LONFRAME = np.append( - LONFRAME, - np.expand_dims(X_lon, axis=0), - axis=0, - ) # noqa: F821 + if INPUTFRAME is not None: + INPUTFRAME = np.append(INPUTFRAME, np.expand_dims(X_fire, axis=0), axis=0) + LATFRAME = np.append(LATFRAME, np.expand_dims(X_lat, axis=0), axis=0) + LONFRAME = np.append(LONFRAME, np.expand_dims(X_lon, axis=0), axis=0) else: INPUTFRAME = np.expand_dims(X_fire, axis=0) LATFRAME = np.expand_dims(X_lat, axis=0)