1 files changed, 7 insertions, 1 deletions
diff --git a/app/client/modules/samplernn/samplernn.reducer.js b/app/client/modules/samplernn/samplernn.reducer.js
index ce3a549..4758f61 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -10,6 +10,7 @@ const samplernnInitialState = {
   folder_id: 0,
   data: null,
   lossReport: null,
+  results: null,
 }
 
 const samplernnReducer = (state = samplernnInitialState, action) => {
@@ -23,7 +24,12 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
         ...state,
         lossReport: action.lossReport,
       }
-
+    case types.samplernn.load_graph:
+      return {
+        ...state,
+        lossReport: action.lossReport,
+        results: action.results,
+      }
     default:
       return state
   }