Use FastJSON instead

* fix pandas warning in latest version
assume-framework · Sep 22, 2023 · 7c96fec · 7c96fec
1 parent 1c74658
commit 7c96fec
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/assume/common/mango_serializer.py b/assume/common/mango_serializer.py
@@ -1,6 +1,6 @@
 import pickle
 
-from mango.messages.codecs import JSON, GenericProtoMsg
+from mango.messages.codecs import FastJSON, GenericProtoMsg
 
 
 def generic_json_serializer():
@@ -29,6 +29,6 @@ def __fromstring__(data):
 
 
 def mango_codec_factory():
-    codec = JSON()
+    codec = FastJSON()
     codec.add_serializer(*generic_json_serializer())
     return codec
diff --git a/assume/common/outputs.py b/assume/common/outputs.py
@@ -317,8 +317,9 @@ def write_market_dispatch(self, data):
         :type data: any
         """
         df = pd.DataFrame(data, columns=["datetime", "power", "market_id", "unit_id"])
-        df["simulation"] = self.simulation_id
-        self.write_dfs["market_dispatch"].append(df)
+        if not df.empty:
+            df["simulation"] = self.simulation_id
+            self.write_dfs["market_dispatch"].append(df)
 
     def write_unit_dispatch(self, data):
         """