pretty_log

BWRC-AMS-ML-Discovery · Nov 29, 2023 · a6d61f7 · a6d61f7
1 parent 94f8348
commit a6d61f7
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/AutoCkt/ML/autockt_ml/autockt_gym.py b/AutoCkt/ML/autockt_ml/autockt_gym.py
@@ -1,4 +1,5 @@
 import datetime
+import json
 from pathlib import Path
 from dataclasses import asdict
 
@@ -136,8 +137,9 @@ def step(self, action):
             "ideal_output_norm": ideal_norm,
             "observation": observation,
         }
+        pretty_log = json.dumps(log, indent=4)
         with open(self.log_file, "a") as f:
-            f.write(f"{log}\n")
+            f.write(f"{pretty_log}\n")
 
         self.num_steps += 1