From ea09b9d9ffd6e80c8747d934c47590c6155de23a Mon Sep 17 00:00:00 2001 From: James McManus Date: Mon, 20 Nov 2023 15:45:56 -0500 Subject: [PATCH] Added fix to defining nowcast_source for tropical runs --- src/common/pg_impl.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/src/common/pg_impl.py b/src/common/pg_impl.py index 6494368..8b75c36 100644 --- a/src/common/pg_impl.py +++ b/src/common/pg_impl.py @@ -288,7 +288,15 @@ def get_station_data(self, **kwargs) -> str: end_date = forecast_data['time_stamp'].iloc[-1] # get nowcast data_source from forecast data_source - nowcast_source = 'NOWCAST_' + "_".join(kwargs['data_source'].split('_')[1:]) + # check if data_source is tropical + if kwargs['data_source'][:2] == 'al': + # if tropical split data source and replace second value (OFCL) with NOWCAST + source_parts = kwargs['data_source'].split('_') + source_parts[1] = 'NOWCAST' + nowcast_source = "_".join(source_parts) + else: + # if synoptic split data source and replace fist value (GFSFORECAST) with NOWCAST + nowcast_source = 'NOWCAST_' + "_".join(kwargs['data_source'].split('_')[1:]) # get obs and nowcast data obs_data = self.get_obs_station_data(kwargs['station_name'], start_date, end_date, nowcast_source)