gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-20 05:52:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix evaluation scoring bug (numpy casts generator to a single-element array containing the generator, which is a truthy object)

Browse Source
This commit is contained in:
Jie Tang
2016-10-20 19:39:15 -07:00
parent 88f94587a2
commit a8e4734cbc
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
gym/scoreboard/scoring.py
View File

@@ -47,7 +47,8 @@ def score_from_merged(episode_lengths, episode_rewards, episode_types, timestamp
"""
if episode_types is not None:
# Select only the training episodes
t_idx = np.where(e == 't' for e in episode_types)
episode_types = np.array(episode_types)
t_idx = np.where(episode_types == 't')
episode_lengths = np.array(episode_lengths)[t_idx]
episode_rewards = np.array(episode_rewards)[t_idx]
timestamps = np.array(timestamps)[t_idx]