From 1268cb78018ce2b13a8d8450afdb6e4f3806c5f0 Mon Sep 17 00:00:00 2001 From: Abodi Massarwa Date: Thu, 8 Aug 2024 22:52:54 +0300 Subject: [PATCH] =?UTF-8?q?final=20commit=F0=9F=99=8F=F0=9F=8F=BD?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../heterogeneous_matroid_constraints_algorithms.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py b/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py index e667113..cc8d695 100644 --- a/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py +++ b/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py @@ -21,8 +21,7 @@ import base64 logger = logging.getLogger(__name__) -# Create a string stream to capture logs -log_stream = io.StringIO() + def per_category_round_robin(alloc: AllocationBuilder, item_categories: dict[str,list], agent_category_capacities: dict[str,dict[str,int]], initial_agent_order: list,callback:callable=None): @@ -1272,13 +1271,16 @@ def helper_generate_bipartite_graph_base64(graph,iteration:int,category:str): return base64.b64encode(img_bytes.read()).decode('utf-8') def helper_configure_logger(): + # Create a string stream to capture logs + log_stream = io.StringIO() stream_handler = logging.StreamHandler(log_stream) formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s') logger.addHandler(stream_handler) stream_handler.setFormatter(formatter) logger.setLevel(logging.DEBUG) + return log_stream -def helper_get_logs(): +def helper_get_logs(log_stream): return log_stream.getvalue() if __name__ == "__main__":