From 445b16848aa24299b27176fe88411b7825f32f53 Mon Sep 17 00:00:00 2001
From: Florian Maurer <maurer@fh-aachen.de>
Date: Mon, 25 Nov 2024 16:26:01 +0100
Subject: [PATCH] fix output moving to utils

---
 assume/common/outputs.py | 2 +-
 assume/world.py          | 8 ++------
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/assume/common/outputs.py b/assume/common/outputs.py
index d827d099..3dd08ac3 100644
--- a/assume/common/outputs.py
+++ b/assume/common/outputs.py
@@ -247,7 +247,7 @@ def handle_output_message(self, content: dict, meta: MetaDict):
             self.write_flows(content_data)
 
         # # keep track of the memory usage of the data
-        self.current_dfs_size += self.calculate_content_size(content_data)
+        self.current_dfs_size += calculate_content_size(content_data)
         # if the current size is larger than self.max_dfs_size, store the data
         if self.current_dfs_size > self.max_dfs_size:
             logger.debug("storing output data due to size limit")
diff --git a/assume/world.py b/assume/world.py
index f57365f0..74c30e2a 100644
--- a/assume/world.py
+++ b/assume/world.py
@@ -198,14 +198,10 @@ def setup(
         container_kwargs = {"mp_method": "fork"} if sys.platform == "linux" else {}
         if self.addr == "world":
             container_func = create_ec_container
-            container_kwargs.update({
-                "addr": self.addr,
-            })
+            container_kwargs.update({"addr": self.addr})
         elif isinstance(self.addr, tuple):
             container_func = create_tcp_container
-            container_kwargs.update({
-                "addr": self.addr,
-            })
+            container_kwargs.update({"addr": self.addr})
         else:
             container_func = create_mqtt_container
             container_kwargs = {