Fix outputs function (#208)

-rl params were not written when not in learning -should be for eval -error handling was wrong --------- Co-authored-by: Florian Maurer <[email protected]>
assume-framework · Sep 26, 2023 · ac75b59 · ac75b59
1 parent e913958
commit ac75b59
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/assume/common/outputs.py b/assume/common/outputs.py
@@ -359,6 +359,9 @@ async def on_stop(self):
                 df = pd.read_sql(query, self.db)
             except (OperationalError, ProgrammingError):
                 continue
+            except Exception as e:
+                logger.error("could not read query: %s", e)
+                continue
 
             dfs.append(df)
 

diff --git a/assume/common/units_operator.py b/assume/common/units_operator.py
@@ -397,6 +397,7 @@ def write_learning_to_output(self, start: datetime, marketconfig: MarketConfig):
 
         db_aid = self.context.data_dict.get("learning_output_agent_id")
         db_addr = self.context.data_dict.get("learning_output_agent_addr")
+
         if db_aid and db_addr:
             self.context.schedule_instant_acl_message(
                 receiver_id=db_aid,

diff --git a/assume/world.py b/assume/world.py
@@ -234,6 +234,8 @@ def add_unit_operator(
             unit_operator_agent._role_context.data_dict = {
                 "output_agent_addr": self.output_agent_addr[0],
                 "output_agent_id": self.output_agent_addr[1],
+                "learning_output_agent_addr": self.output_agent_addr[0],
+                "learning_output_agent_id": self.output_agent_addr[1],
             }
 
     async def async_add_unit(