Skip to content

Commit 3a353c4

Browse files
improve logging for sorting (#329)
* improve logging for sorting * dont sort if there are any forecasts * fix * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * tidy --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 06cc402 commit 3a353c4

File tree

1 file changed

+11
-4
lines changed

1 file changed

+11
-4
lines changed

nowcasting_datamodel/read/read.py

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -225,9 +225,18 @@ def sort_all_forecast_value(forecasts: List[ForecastSQL]):
225225
:param forecasts: list of forecasts
226226
:return: list of forecasts, but with sorted forecat values
227227
"""
228-
""" Sorting all forecasts"""
229228

230-
logger.debug("sorting 'forecast_values_latest' or 'forecast_values' values.")
229+
if len(forecasts) == 0:
230+
return forecasts
231+
232+
if forecasts[0].historic:
233+
logger.debug("Sorting 'forecast_values_latest' values")
234+
logger.debug(
235+
f"Will be sorting {len(forecasts[0].forecast_values_latest)} " f"forecast values latest"
236+
)
237+
else:
238+
logger.debug("Sorting 'forecast_values' values")
239+
logger.debug(f"Will be sorting {len(forecasts[0].forecast_values)} " f"forecast values")
231240

232241
for forecast in forecasts:
233242
sort_forecast_values(forecast=forecast)
@@ -374,8 +383,6 @@ def get_latest_forecast_for_gsps(
374383
forecasts = query.all()
375384

376385
logger.debug(f"Found {len(forecasts)} forecasts")
377-
if len(forecasts) > 0:
378-
logger.debug(f"The first forecast has {len(forecasts[0].forecast_values)} forecast_values")
379386

380387
forecasts = sort_all_forecast_value(forecasts)
381388

0 commit comments

Comments
 (0)