sims running, need csv write

experiments/dutch_auctions/experiment.py

@@ -4,6 +4,7 @@
 from copy import deepcopy
 import matplotlib.pyplot as plt
 
+from pydss.pymaker.numeric import Wad, Rad, Ray
 from pydss.util import RequireException
 from experiments.experiment import Experiment
 
@@ -143,6 +144,10 @@ def run(self, sim_name):
 
             historical_stats.append(deepcopy(state["stats"]))
 
+            self.write(
+                f"/bab-stablesims/experiments/dutch_auctions/results/{sim_name}_state.json", state, t
+            )
+
         _, axs = plt.subplots(4)
         time_range = list(range(self.parameters["timesteps"]))
         plt.title(f"Chip: {float(state['clippers']['WETH'].chip)}, Tip: {float(state['clippers']['WETH'].tip)}")
@@ -151,16 +156,13 @@ def run(self, sim_name):
         axs[2].plot(time_range, [stats["num_unsafe_vaults"] for stats in historical_stats])
         axs[3].plot(time_range, [stats["incentive_amount"] for stats in historical_stats])
         plt.savefig(f"/bab-stablesims/experiments/dutch_auctions/results/{sim_name}.png")
-        self.write(
-            f"/bab-stablesims/experiments/dutch_auctions/results/{sim_name}_state.json", state
-        )
 
     def format_data(self, state, full_state=True):
         data = state if full_state else state["stats"]
         data = deepcopy(data)
         for key, value in data.items():
             if isinstance(value, (Ray, Rad, Wad)):
-                data[key] = float(data[value])
+                data[key] = float(value)
             elif isinstance(value, dict):
                 data[key] = self.format_data(value)
             elif hasattr(value, "__iter__"):
@@ -172,4 +174,4 @@ def write(self, filename, data, t):
         with open(filename, "a") as f:
             f.write("==================\n")
             f.write("Timestep: {}".format(t))
-            f.write(self.format_data(data) + "\n")
+            f.write(str(self.format_data(data, False)) + "\n")