Some exception handling in plotter.

openai · Mar 14, 2019 · 41ce375 · 41ce375
1 parent 8b92b8a
commit 41ce375
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/spinup/utils/plot.py b/spinup/utils/plot.py
@@ -84,7 +84,11 @@ def get_datasets(logdir, condition=None):
             unit = units[condition1]
             units[condition1] += 1
 
-            exp_data = pd.read_table(os.path.join(root,'progress.txt'))
+            try:
+                exp_data = pd.read_table(os.path.join(root,'progress.txt'))
+            except:
+                print('Could not read from %s'%os.path.join(root,'progress.txt'))
+                continue
             performance = 'AverageTestEpRet' if 'AverageTestEpRet' in exp_data else 'AverageEpRet'
             exp_data.insert(len(exp_data.columns),'Unit',unit)
             exp_data.insert(len(exp_data.columns),'Condition1',condition1)