From a25eff9537ad95158818e7a074bfe2f2630d630b Mon Sep 17 00:00:00 2001
From: flaurent <florian.laurent@gmail.com>
Date: Mon, 7 Sep 2020 03:43:55 +0200
Subject: [PATCH] Better logging

---
 flatland/evaluators/service.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/flatland/evaluators/service.py b/flatland/evaluators/service.py
index 8b15b58c..2cf4845b 100644
--- a/flatland/evaluators/service.py
+++ b/flatland/evaluators/service.py
@@ -651,7 +651,7 @@ class FlatlandRemoteEvaluationService:
             mean_test_complete_percentage = np.mean(self.simulation_percentage_complete_per_test[self.current_test])
             if mean_test_complete_percentage < TEST_MIN_PERCENTAGE_COMPLETE_MEAN:
                 print("=" * 15)
-                print("Mean percentage done too low: {} < {}. Evaluation will stop here.".format(
+                print("Mean percentage done too low: {:.3f} < {}. Evaluation will stop here.".format(
                     mean_test_complete_percentage,
                     TEST_MIN_PERCENTAGE_COMPLETE_MEAN
                 ))
@@ -870,6 +870,8 @@ class FlatlandRemoteEvaluationService:
                     self.nb_malfunctioning_trains[-1]
                 ))
 
+            print("Total normalized reward so far: {:.3f}".format(sum(self.simulation_rewards_normalized)))
+
             # Write intermediate results
             if self.result_output_path:
                 self.evaluation_metadata_df.to_csv(self.result_output_path)
-- 
GitLab