2 files changed, 43 insertions, 0 deletions
diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js
new file mode 100644
index 0000000..fa41fe8
--- /dev/null
+++ b/app/client/modules/samplernn/samplernn.tasks.js
@@ -0,0 +1,37 @@
+import uuidv1 from 'uuid/v1'
+
+import socket from '../../socket'
+import types from '../../types'
+
+import actions from '../../actions'
+
+export const train_task = (dataset, epochs=1) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'train',
+    dataset: dataset,
+    epochs: epochs,
+    opt: {
+      sample_length: 44100 * 5,
+      n_samples: 6,
+      keep_old_checkpoints: false,
+    }
+  }
+  return actions.queue.add_task(task)
+}
+export const fetch_task = (dataset) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'fetch',
+    dataset: dataset,
+  }
+  return actions.queue.add_task(task)
+}
+export const clear_cache_task = (dataset) => dispatch => {
+  const task = {
+    module: 'samplernn',
+    activity: 'clear_cache',
+    dataset: dataset,
+  }
+  return actions.queue.add_task(task)
+}
+\ No newline at end of file
diff --git a/app/client/socket/socket.task.js b/app/client/socket/socket.task.js
index 2f3c65a..3a78c6c 100644
--- a/app/client/socket/socket.task.js
+++ b/app/client/socket/socket.task.js
@@ -40,6 +40,12 @@ socket.on('task_res', (data) => {
       break
     case 'set_priority':
       break
+    case 'progress':
+      dispatch({ type: types.task.progress, task: data.task })
+      break
+    case 'epoch':
+      dispatch({ type: types.task.progress, task: data.task })
+      break
     case 'error':
       return console.log('task error', data)
     default: