must be string

2 files changed, 14 insertions, 10 deletions

diff --git a/app/client/modules/samplernn/samplernn.tasks.js b/app/client/modules/samplernn/samplernn.tasks.js
index f253047..ee26347 100644
--- a/app/client/modules/samplernn/samplernn.tasks.js
+++ b/app/client/modules/samplernn/samplernn.tasks.js
@@ -10,7 +10,7 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
     module: 'samplernn',
     activity: 'train',
     dataset: dataset.name,
-    epoch: dataset.checkpoints.length ? dataset.checkpoints[0].epoch || 0 : 0
+    epoch: dataset.checkpoints.length ? dataset.checkpoints[0].epoch || 0 : 0,
     epochs: epochs,
     opt: {
       folder_id: folder_id,
diff --git a/app/client/modules/samplernn/views/samplernn.show.js b/app/client/modules/samplernn/views/samplernn.show.js
index e74de4f..45235b1 100644
--- a/app/client/modules/samplernn/views/samplernn.show.js
+++ b/app/client/modules/samplernn/views/samplernn.show.js
@@ -7,11 +7,10 @@ import * as samplernnActions from '../samplernn.actions'
 import * as samplernnTasks from '../samplernn.tasks'
 import * as audioPlayerActions from '../../../common/audioPlayer/audioPlayer.actions'
 
-import Loading from '../../../common/loading.component'
+import { Loading, CurrentTask, FileList, FileRow } from '../../../common'
+
 import DatasetForm from '../../../dataset/dataset.form'
-import NewDatasetForm from '../../../dataset/dataset.new'
 import UploadStatus from '../../../dataset/upload.status'
-import { FileList, FileRow } from '../../../common/fileList.component'
 
 import DatasetComponent from '../../../dataset/dataset.component'
 
@@ -38,17 +37,16 @@ class SampleRNNShow extends Component {
     }
   }
   render(){
-    const { samplernn, match, history } = this.props
+    const { samplernn, runner, match, history } = this.props
     const { folderLookup } = (samplernn.data || {})
+    console.log(runner)
     const folder = (folderLookup || {})[samplernn.folder_id] || {}
     return (
       <div className='app samplernn'>
-        <div class='heading'>
-          <div class='spaced'>
-            <h1>{folder ? folder.name : <Loading />}</h1>
-            <UploadStatus />
-          </div>
+        <div className='heading'>
+          <h1>{folder ? folder.name : <Loading />}</h1>
         </div>
+        <div className='row'>
         {folder && folder.name && folder.name !== 'unsorted' &&
           <DatasetForm
             title='Add Files'
@@ -57,6 +55,11 @@ class SampleRNNShow extends Component {
             canUpload canAddURL
           />
         }
+          <div>
+            <UploadStatus />
+            <CurrentTask processor='gpu' />
+          </div>
+        </div>
         <DatasetComponent
           loading={samplernn.loading}
           progress={samplernn.progress}
@@ -114,6 +117,7 @@ class SampleRNNShow extends Component {
 
 const mapStateToProps = state => ({
   samplernn: state.module.samplernn,
+  runner: state.system.runner,
 })
 
 const mapDispatchToProps = (dispatch, ownProps) => ({