huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Close figures

Browse Source
This commit is contained in:
ebuehrle
2022-03-04 16:14:35 +01:00
parent 575e299fc8
commit a576f0fb18
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/evaluation/evaluation.py
View File

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