ledmands commited on
Commit
44830bc
1 Parent(s): 2ef57f9

Modified plot_improvement.py to fix agent directory sorting bug

Browse files
Files changed (1) hide show
  1. plot_improvement.py +3 -1
plot_improvement.py CHANGED
@@ -19,7 +19,9 @@ def calc_stats(filepath):
19
  # args = parser.parse_args()
20
 
21
  filepaths = []
22
- for d in os.listdir("agents/"):
 
 
23
  if "dqn_v2" in d:
24
  path = "agents/" + d + "/evaluations.npz"
25
  filepaths.append(path)
 
19
  # args = parser.parse_args()
20
 
21
  filepaths = []
22
+ agent_dirs = os.listdir("agents/")
23
+ agent_dirs.sort()
24
+ for d in agent_dirs:
25
  if "dqn_v2" in d:
26
  path = "agents/" + d + "/evaluations.npz"
27
  filepaths.append(path)