From b99c83c8baece8c5012d6297cea6802a3a45576f Mon Sep 17 00:00:00 2001 From: "Dylan H. Morris" Date: Sat, 28 Dec 2024 18:05:39 -0500 Subject: [PATCH] Test both diseases in end to end test --- pipelines/tests/test_end_to_end.sh | 33 ++++++++++++++++-------------- 1 file changed, 18 insertions(+), 15 deletions(-) diff --git a/pipelines/tests/test_end_to_end.sh b/pipelines/tests/test_end_to_end.sh index ca09651f..c6b20501 100755 --- a/pipelines/tests/test_end_to_end.sh +++ b/pipelines/tests/test_end_to_end.sh @@ -15,23 +15,26 @@ if [ $? -ne 0 ]; then else echo "TEST-MODE: Finished generating test data" fi -echo "TEST-MODE: Running forecasting pipeline for several locations" +echo "TEST-MODE: Running forecasting pipeline for two diseases in multiple locations" for state in CA MT US do - python pipelines/forecast_state.py \ - --disease "COVID-19" \ - --state $state \ - --facility-level-nssp-data-dir "$BASE_DIR/private_data/nssp_etl_gold" \ - --state-level-nssp-data-dir "$BASE_DIR/private_data/nssp_state_level_gold" \ - --priors-path "$BASE_DIR/test_output/priors.py" \ - --param-data-dir "$BASE_DIR/private_data/prod_param_estimates" \ - --output-dir "$BASE_DIR/private_data" \ - --n-training-days 60 \ - --n-chains 2 \ - --n-samples 250 \ - --n-warmup 250 \ - --score \ - --eval-data-path "$BASE_DIR/private_data/nssp-archival-vintages" + for disease in COVID-19 Influenza + do + python pipelines/forecast_state.py \ + --disease $disease \ + --state $state \ + --facility-level-nssp-data-dir "$BASE_DIR/private_data/nssp_etl_gold" \ + --state-level-nssp-data-dir "$BASE_DIR/private_data/nssp_state_level_gold" \ + --priors-path "$BASE_DIR/test_output/priors.py" \ + --param-data-dir "$BASE_DIR/private_data/prod_param_estimates" \ + --output-dir "$BASE_DIR/private_data" \ + --n-training-days 60 \ + --n-chains 2 \ + --n-samples 250 \ + --n-warmup 250 \ + --score \ + --eval-data-path "$BASE_DIR/private_data/nssp-archival-vintages" + done done if [ $? -ne 0 ]; then