1 files changed, 10 insertions, 0 deletions
diff --git a/app/client/dataset/dataset.reducer.js b/app/client/dataset/dataset.reducer.js
index 10b4a94..f303a7f 100644
--- a/app/client/dataset/dataset.reducer.js
+++ b/app/client/dataset/dataset.reducer.js
@@ -8,6 +8,7 @@ import types from '../types'
 
 const datasetInitialState = () => ({
   loading: true,
+  progress: null,
   error: null,
   data: null,
   folder_id: 0,
@@ -57,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
   }