From 4a3d06ec726bfb4c50a0f588d81d320b68386370 Mon Sep 17 00:00:00 2001
From: Alexey Skrynnik <Tviskaron@gmail.com>
Date: Sun, 29 Oct 2023 16:40:00 +0300
Subject: [PATCH] fixed metrics logging for PyMARL integration with lifelong
 scenario

---
 pogema/integrations/pymarl.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pogema/integrations/pymarl.py b/pogema/integrations/pymarl.py
index 9421877..5a9ec87 100644
--- a/pogema/integrations/pymarl.py
+++ b/pogema/integrations/pymarl.py
@@ -28,9 +28,9 @@ def step(self, actions):
         info = {}
         done = all(terminated) or all(truncated)
         if done:
-            info.update(CSR=infos[0]['metrics']['CSR'])
-            info.update(ISR=infos[0]['metrics']['ISR'])
-
+            for key, value in infos[0]['metrics'].items():
+                info[key] = value
+            
         return sum(rewards), done, info
 
     def get_obs(self):