Skip to content

Commit 971cbea

Browse files
committed
adding extra forecasters
1 parent c245485 commit 971cbea

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

scripts/get_forecast_data.r

+8-3
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,8 @@
1313
# 4. TODO: Add a companion metadata file that describes the forecasts in the parquet file
1414
# 5. TODO: Add the metadata file to the S3 bucket
1515
# 6. TODO: Allow incremental updates to the forecasts, by downloading only the new files
16-
16+
print("-------------------------------------------------------")
17+
print(glue::glue("starting at {Sys.time()}"))
1718
suppressPackageStartupMessages({
1819
library(tidyverse)
1920
library(httr)
@@ -32,15 +33,15 @@ run_time_local <- with_tz(run_time)
3233
# Configuration
3334
covid_config <- list(
3435
base_url = "https://raw.githubusercontent.com/cdcgov/covid19-forecast-hub/main/model-output",
35-
forecasters = c("CMU-TimeSeries", "CovidHub-baseline", "CovidHub-ensemble"),
36+
forecasters = c("CMU-TimeSeries", "CovidHub-baseline", "CovidHub-ensemble", "UMass-ar6_pooled", "UMass-gbqr", "CEPH-Rtrend_covid", "Metaculus-cp"),
3637
s3_bucket = "forecasting-team-data",
3738
s3_key = "exploration/2024-2025_covid_hosp_forecasts.parquet",
3839
disease = "covid"
3940
)
4041
# same but for flu
4142
flu_config <- list(
4243
base_url = "https://raw.githubusercontent.com/cdcepi/FluSight-forecast-hub/main/model-output",
43-
forecasters = c("FluSight-baseline", "FluSight-ensemble", "CMU-TimeSeries"),
44+
forecasters = c("FluSight-baseline", "FluSight-ensemble", "CMU-TimeSeries", "PSI-PROF", "FluSight-lop_norm", "UMass-flusion", "NIH-Flu_ARIMA", "Metaculus-cp"),
4445
s3_bucket = "forecasting-team-data",
4546
s3_key = "exploration/2024-2025_flu_hosp_forecasts.parquet",
4647
disease = "flu"
@@ -152,3 +153,7 @@ cli::cli_alert_info("Fetching COVID forecasts {run_time_local} (UTC: {run_time})
152153
fetch_forecast_files(sync_to_s3 = FALSE, disease = "covid")
153154
cli::cli_alert_info("Fetching FLU forecasts {run_time_local} (UTC: {run_time})")
154155
fetch_forecast_files(sync_to_s3 = FALSE, disease = "flu")
156+
157+
print(glue::glue("Run successfully finished at {Sys.time()}"))
158+
print("-------------------------------------------------------")
159+
print("-------------------------------------------------------")

0 commit comments

Comments
 (0)