fix bug

RL4H · Jun 6, 2024 · 22a9620 · 22a9620
1 parent b51bdb2
commit 22a9620
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments/run_clinical_treatment.py b/experiments/run_clinical_treatment.py
@@ -55,7 +55,7 @@ def run_simulation(args, id=0, rollout_steps=288*30, n_trials=10, seed=0):
                 bolus_carbs = carb_estimate(bolus_carbs, info['day_hour'], patients[id], type=args.carb_estimation_method)
                 #print('The real carbs : {}, estimated carbs: {}'.format(real_bolus, bolus_carbs))
 
-            trial_history[counter] = [next_state.CGM, carbs, action, counter]
+            trial_history[counter] = [next_state.CGM, carbs, action[0], counter]
 
             counter += 1
             action = controller.get_action(meal=bolus_carbs, glucose=next_state.CGM)  # BB Controller.