ledmands
commited on
Commit
·
44830bc
1
Parent(s):
2ef57f9
Modified plot_improvement.py to fix agent directory sorting bug
Browse files- 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 |
-
|
|
|
|
|
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)
|