Skip to content

Commit 63b571a

Browse files
filter by model id in forecast value lates table (#186)
* filter by model id in forecast value lates table * fix * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 97cc014 commit 63b571a

File tree

1 file changed

+11
-5
lines changed

1 file changed

+11
-5
lines changed

nowcasting_datamodel/read/read.py

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -309,11 +309,6 @@ def get_latest_forecast_for_gsps(
309309
# filter on historic
310310
query = query.filter(ForecastSQL.historic == historic)
311311

312-
# filter on model name
313-
if model_name is not None:
314-
query = query.join(MLModelSQL)
315-
query = query.filter(MLModelSQL.name == model_name)
316-
317312
# filter on target time
318313
if start_target_time is not None:
319314
query = filter_query_on_target_time(
@@ -323,6 +318,17 @@ def get_latest_forecast_for_gsps(
323318
end_target_time=end_target_time,
324319
)
325320

321+
# filter on model name
322+
if model_name is not None:
323+
if historic:
324+
# if start target time is None, we need to join with forecast value latest
325+
if start_target_time is None:
326+
query = query.join(ForecastValueLatestSQL)
327+
query = query.join(MLModelSQL, ForecastValueLatestSQL.model_id == MLModelSQL.id)
328+
else:
329+
query = query.join(MLModelSQL)
330+
query = query.filter(MLModelSQL.name == model_name)
331+
326332
query = query.join(LocationSQL)
327333

328334
# option to preload values, makes querying quicker

0 commit comments

Comments
 (0)