fix unset reward

assume-framework · Oct 31, 2023 · 7482680 · 7482680
1 parent 2fc1497
commit 7482680
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/assume/common/outputs.py b/assume/common/outputs.py
@@ -411,8 +411,10 @@ def get_sum_reward(self):
             f"select reward FROM rl_params where simulation='{self.simulation_id}'"
         )
 
+        avg_reward = 0
         with self.db.begin() as db:
             reward = db.execute(query).fetchall()
+        if len(reward):
             avg_reward = sum(r[0] for r in reward) / len(reward)
 
         return avg_reward