From fe48736dfe8e37e9fe09f5afebe781c71bdfaf40 Mon Sep 17 00:00:00 2001
From: cbernalz <cbernalz@uci.edu>
Date: Mon, 25 Nov 2024 15:09:25 -0800
Subject: [PATCH] 2024-11-25 update : fixing hosp only model.

---
 src/uciwweihr_gq_pp.jl | 28 ++++++++++++++--------------
 1 file changed, 14 insertions(+), 14 deletions(-)

diff --git a/src/uciwweihr_gq_pp.jl b/src/uciwweihr_gq_pp.jl
index 8e12bed..8d9444d 100644
--- a/src/uciwweihr_gq_pp.jl
+++ b/src/uciwweihr_gq_pp.jl
@@ -109,6 +109,7 @@ function uciwweihr_gq_pp(
     return(results)
 end
 
+
 function uciwweihr_gq_pp(
     samples,
     data_hosp,
@@ -120,6 +121,7 @@ function uciwweihr_gq_pp(
     return_bool::Bool=true,
     gq_bool::Bool=true
     )
+
     println("Using uciwweihr_model without wastewater!!!")
     obstimes_hosp = convert(Vector{Float64}, obstimes_hosp)
     param_change_times = convert(Vector{Float64}, param_change_times)
@@ -139,11 +141,11 @@ function uciwweihr_gq_pp(
     end
 
     my_model = uciwweihr_model(
-        data_hosp,
+        data_hosp, 
         obstimes_hosp;
         param_change_times,
         params,
-        return_bool
+        return_bool# = true
     )
 
 
@@ -154,24 +156,21 @@ function uciwweihr_gq_pp(
     #gq_randn = Chains(generated_quantities(my_model, samples_randn))
 
     my_model_forecast_missing = uciwweihr_model(
-        missing_data_hosp,
+        missing_data_hosp, 
         obstimes_hosp;
         param_change_times,
         params,
-        return_bool
+        return_bool# = true,
     )
 
-
-    indices_to_keep = .!isnothing.(generated_quantities(my_model, samples))
-    samples_randn = ChainsCustomIndex(samples, indices_to_keep)
-
-
-    Random.seed!(seed)
-    predictive_randn = predict(my_model_forecast_missing, samples_randn)
-
     samples_df = DataFrame(samples)
 
     if gq_bool
+        indices_to_keep = .!isnothing.(generated_quantities(my_model, samples))
+        samples_randn = ChainsCustomIndex(samples, indices_to_keep)
+    
+        Random.seed!(seed)
+        predictive_randn = predict(my_model_forecast_missing, samples_randn)
         Random.seed!(seed)
         println("Generating quantities...")
         gq_randn = Chains(generated_quantities(my_model, samples_randn))
@@ -179,9 +178,10 @@ function uciwweihr_gq_pp(
     else
         println("Not generating quantities...")
         println("**result will only contain pp**")
+        Random.seed!(seed)
+        predictive_randn = predict(my_model_forecast_missing, samples)
         results = [DataFrame(predictive_randn)]
-        
     end
 
     return(results)
-end
\ No newline at end of file
+end