add policy market simulation

assume-framework · Nov 10, 2023 · be054d1 · be054d1
1 parent 27d1ff0
commit be054d1
Show file tree

Hide file tree

Showing 5 changed files with 1,019 additions and 0 deletions.
diff --git a/assume/common/outputs.py b/assume/common/outputs.py
@@ -299,6 +299,12 @@ def write_market_orders(self, market_orders, market_id):
 
         market_orders = separate_orders(market_orders)
         df = pd.DataFrame.from_records(market_orders, index="start_time")
+        if "eligible_lambda" in df.columns:
+            df["eligible_lambda"] = df["eligible_lambda"].apply(lambda x: x.__name__)
+        if "evaluation_frequency" in df.columns:
+            df["evaluation_frequency"] = df["evaluation_frequency"].apply(
+                lambda x: repr(x)
+            )
 
         del df["only_hours"]
         del df["agent_id"]