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 27d4d58..372b3bc 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -6,12 +6,12 @@ const samplernnInitialState = {
   folders: [],
   folder: {},
   data: null,
+  lossReport: null,
 }
 
 const samplernnReducer = (state = samplernnInitialState, action) => {
   switch(action.type) {
     case types.samplernn.init:
-      console.log(action.data)
       return {
         ...state,
         loading: false,
@@ -54,6 +54,12 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
         folder: action.folder,
       }
 
+    case types.samplernn.load_loss:
+      return {
+        ...state,
+        lossReport: action.lossReport,
+      }
+
     default:
       return state
   }