Bug fix when "baselines" folder did not exist

KongMingxi · Sep 24, 2018 · 1c0a6e5 · 1c0a6e5
1 parent 2b88851
commit 1c0a6e5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/evaluation/gather_results.py b/evaluation/gather_results.py
@@ -66,8 +66,9 @@ def computeStates(exp_config, log_dir):
     if 'baselines' not in item and os.path.isdir('{}/{}'.format(log_dir, item)):
         experiments.append(item)
 
-for item in os.listdir(log_dir + "/baselines"):
-    experiments.append('baselines/' + item)
+if os.path.isdir(log_dir + "/baselines"):
+    for item in os.listdir(log_dir + "/baselines"):
+        experiments.append('baselines/' + item)
 
 experiments.sort()
 print("Found {} experiments".format(len(experiments)))