pix2pix datasets in full effect

1 files changed, 14 insertions, 1 deletions

diff --git a/app/client/dataset/dataset.reducer.js b/app/client/dataset/dataset.reducer.js
index 287ebb8..d7951d0 100644
--- a/app/client/dataset/dataset.reducer.js
+++ b/app/client/dataset/dataset.reducer.js
@@ -10,17 +10,30 @@ const datasetInitialState = () => ({
   loading: true,
   error: null,
   data: null,
+  folder_id: 0,
 })
 
 const datasetReducer = (state = datasetInitialState(), action) => {
   switch(action.type) {
+    case types.app.load_progress:
+      return {
+        ...state,
+        progress: action.progress
+      }
+
     case types.dataset.load:
       return {
         ...state,
         loading: false,
         data: action.data,
       }
-    
+
+    case types.dataset.set_folder:
+      return {
+        ...state,
+        folder_id: action.data.folder_id,
+      }
+
     case types.folder.create:
       if (action.data.module === 'samplernn') {
         return {