Resolved some warnings and style issues

marl_factory_grid/utils/logging/envmonitor.py

@@ -9,7 +9,7 @@ from marl_factory_grid.utils.helpers import IGNORED_DF_COLUMNS
 
 import pandas as pd
 
-from marl_factory_grid.utils.plotting.compare_runs import plot_single_run
+from marl_factory_grid.utils.plotting.plot_single_runs import plot_single_run
 
 
 class EnvMonitor(Wrapper):
@@ -22,7 +22,6 @@ class EnvMonitor(Wrapper):
         self._monitor_df = pd.DataFrame()
         self._monitor_dict = dict()
 
-
     def step(self, action):
         obs_type, obs, reward, done, info = self.env.step(action)
         self._read_info(info)

marl_factory_grid/utils/logging/recorder.py

@@ -2,11 +2,9 @@ from os import PathLike
 from pathlib import Path
 from typing import Union, List
 
-import yaml
-from gymnasium import Wrapper
-
 import numpy as np
 import pandas as pd
+from gymnasium import Wrapper
 
 
 class EnvRecorder(Wrapper):
@@ -106,7 +104,7 @@ class EnvRecorder(Wrapper):
                 out_dict = {'episodes': self._recorder_out_list}
             out_dict.update(
                 {'n_episodes': self._curr_episode,
-                 'metadata':dict(
+                 'metadata': dict(
                      level_name=self.env.params['General']['level_name'],
                      verbose=False,
                      n_agents=len(self.env.params['Agents']),