3 files changed, 7 insertions, 7 deletions
diff --git a/app/client/modules/samplernn/samplernn.reducer.js b/app/client/modules/samplernn/samplernn.reducer.js
index e70be5b..ed602ba 100644
--- a/app/client/modules/samplernn/samplernn.reducer.js
+++ b/app/client/modules/samplernn/samplernn.reducer.js
@@ -56,7 +56,7 @@ const samplernnReducer = (state = samplernnInitialState, action) => {
       console.log(action)
       return state
     case types.socket.status:
-      return samplernnSocket(state, action)
+      return samplernnSocket(state, action.data)
     default:
       return state
   }
diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js
index 601acbb..01a3b82 100644
--- a/app/relay/modules/samplernn.js
+++ b/app/relay/modules/samplernn.js
@@ -6,7 +6,7 @@ const cwd = process.env.SAMPLERNN_CWD || path.join(process.env.HOME, 'code/' + n
 const fetch = {
   type: 'perl',
   script: 'get.pl',
-  build_params: (task) => {
+  params: (task) => {
     console.log(task)
     return [ task.opt.url ]
   }
@@ -14,7 +14,7 @@ const fetch = {
 const train = {
   type: 'pytorch',
   script: 'train.py',
-  build_params: (task) => {
+  params: (task) => {
   },
   onComplete: publish,
   // python train.py \
@@ -28,7 +28,7 @@ const train = {
 const generate = {
   type: 'pytorch',
   script: 'generate.py',
-  build_params: (task) => {
+  params: (task) => {
   },
   onComplete: publish,
   // python generate.py \
@@ -42,7 +42,7 @@ const generate = {
 const publish = {
   type: 'perl',
   script: 'latest.pl',
-  build_params: (task) => {
+  params: (task) => {
   } 
 }
 // after train and generate, run perl latest.pl -l $checkpoint_name
diff --git a/app/relay/runner.js b/app/relay/runner.js
index ecdfcde..2713a2e 100644
--- a/app/relay/runner.js
+++ b/app/relay/runner.js
@@ -65,8 +65,8 @@ export function build_params(module, task) {
 	const activity = module.activities[task.activity]
 	const interpreter = interpreters[activity.type]
   let opt_params;
-	if (activity.build_params) {
-		opt_params = activity.build_params(task)
+	if (typeof activity.params === 'function') {
+		opt_params = activity.params(task)
 	}
 	else {
 		const opt = task.opt