list checkpoint and epoch

1 files changed, 9 insertions, 0 deletions

diff --git a/app/client/dataset/dataset.reducer.js b/app/client/dataset/dataset.reducer.js
index 065d3da..f303a7f 100644
--- a/app/client/dataset/dataset.reducer.js
+++ b/app/client/dataset/dataset.reducer.js
@@ -58,6 +58,15 @@ const datasetReducer = (state = datasetInitialState(), action) => {
     case types.file.destroy:
       return handleFileDestroy(state, action)
 
+    case types.dataset.list_epochs:
+      return {
+        ...state,
+        data: {
+          ...state.data,
+          epochs: action.data.epochs,
+        }
+      }
+
     default:
       return state
   }