pass thru folder id

1 files changed, 5 insertions, 6 deletions

diff --git a/app/client/modules/samplernn/samplernn.show.js b/app/client/modules/samplernn/samplernn.show.js
index 145b784..acc7692 100644
--- a/app/client/modules/samplernn/samplernn.show.js
+++ b/app/client/modules/samplernn/samplernn.show.js
@@ -59,18 +59,17 @@ class SampleRNNShow extends Component {
   }
   datasetActions(dataset, isFetching=false, isProcessing=false){
     const { samplernn, remote } = this.props
-    const { fileLookup } = samplernn.data
     const input = fileLookup[dataset.input[0]]
     if (! input) return null
     if (input.name && input.name.match(/(gif|jpe?g|png)$/i)) return null
     return (
       <div>
         <div class={'actions'}>
-          <span class='link' onClick={() => remote.train_task(dataset, 1)}>train</span>
-          <span class='link' onClick={() => remote.train_task(dataset, 2)}>2x</span>
-          <span class='link' onClick={() => remote.train_task(dataset, 4)}>4x</span>
-          <span class='link' onClick={() => remote.train_task(dataset, 6)}>6x</span>
-          <span class='link' onClick={() => remote.train_task(dataset, 18)}>18x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 1)}>train</span>
+          <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 2)}>2x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 4)}>4x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 6)}>6x</span>
+          <span class='link' onClick={() => remote.train_task(dataset, samplernn.folder_id, 18)}>18x</span>
         </div>
         {dataset.isBuilt
         ? <div class='subtext'>