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): """