From a576f0fb188c40a83408d7db0e86a28c016d1920 Mon Sep 17 00:00:00 2001
From: ebuehrle <43623224+ebuehrle@users.noreply.github.com>
Date: Fri, 4 Mar 2022 16:14:35 +0100
Subject: [PATCH] Close figures

---
 src/evaluation/evaluation.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/evaluation/evaluation.py b/src/evaluation/evaluation.py
index b47f978..559baed 100644
--- a/src/evaluation/evaluation.py
+++ b/src/evaluation/evaluation.py
@@ -10,6 +10,7 @@ from src.util.wrappers import IntersimpleTimeLimit
 from src.options.envs import OptionsEnv
 from src.safe_options.options import SafeOptionsEnv
 from src.evaluation.vec_env import CallbackWhenDoneVecEnv
+import matplotlib.pyplot as plt
 
 class IntersimpleEvaluation:
     """
@@ -156,6 +157,7 @@ class IntersimpleEvaluation:
         agent = info['agent']
         filestr = os.path.join(self.videos_folder, f'agent{agent}')
         self.env.close(filestr=filestr)
+        plt.close('all')
     
     def post_proc(self):
         """