1 files changed, 6 insertions, 3 deletions
diff --git a/app/client/live/reducer.js b/app/client/live/reducer.js
index e14a039..f774d3e 100644
--- a/app/client/live/reducer.js
+++ b/app/client/live/reducer.js
@@ -37,13 +37,16 @@ const liveReducer = (state = liveInitialState, action) => {
       return {
         ...state,
         checkpoints: action.checkpoints,
+        epochs: [],
       }
 
-    case 'LIST_CHECKPOINT_DIR':
+    case 'LIST_EPOCHS':
       return {
         ...state,
-        checkpoint_dir: action.checkpoint_dir,
-      }
+        epochs: action.epochs.map(a => [ a == 'latest' ? Infinity : a, a ])
+                      .sort((a,b) => a[0] - b[0])
+                      .map(a => a[1])
+    }
 
     case 'LIST_SEQUENCES':
       return {