1 files changed, 10 insertions, 0 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.reducer.js b/app/client/modules/pix2pixhd/pix2pixhd.reducer.js
index c3d52a3..5a2afc0 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.reducer.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.reducer.js
@@ -8,6 +8,11 @@ const pix2pixhdInitialState = {
   folder_id: 0,
   data: null,
   results: null,
+  checkpoint: {
+    name: '',
+    sequenceCount: 0,
+    datasetCount: 0,
+  }
 }
 
 const pix2pixhdReducer = (state = pix2pixhdInitialState, action) => {
@@ -21,6 +26,11 @@ const pix2pixhdReducer = (state = pix2pixhdInitialState, action) => {
         ...state,
         results: action.results,
       }
+    case types.pix2pixhd.load_dataset_count:
+      return {
+        ...state,
+        checkpoint: action.data,
+      }
     case types.file.destroy:
       console.log('file destroy', state.results)
       return {