1 files changed, 14 insertions, 5 deletions
diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index 683d44a..c10a1db 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -8,6 +8,7 @@ import * as samplernnActions from './samplernn.actions'
 import DatasetForm from '../../dataset/dataset.form'
 import NewDatasetForm from '../../dataset/dataset.new'
 import { FileList, FileRow } from '../../common/fileList.component'
+import Loading from '../../common/loading.component'
 
 const samplernnModule = {
   name: 'samplernn',
@@ -62,7 +63,8 @@ class SampleRNNDatasets extends Component {
     const { samplernn, match, history } = this.props
     if (samplernn.loading) {
       // console.log('loading')
-      return <span>Loading</span>
+      console.log(samplernn.loading)
+      return <Loading progress={samplernn.progress} />
     }
     if (!samplernn.data.folders.length) {
       console.log('no folders, redirect to /new')
@@ -83,13 +85,13 @@ class SampleRNNDatasets extends Component {
     )
   }
   renderGroups(){
-    const { samplernn, onPickDataset } = this.props
+    const { samplernn, onPickDataset, actions } = this.props
     const folder = samplernn.folder
     const { mapFn, sortFn } = util.sort.orderByFn('date desc')
     const datasets = folder.datasets.map(mapFn).sort(sortFn).map(pair => {
       const dataset = pair[1]
       return (
-        <div className='row dataset' onClick={() => onPickDataset(dataset)}>
+        <div className='row dataset' onClick={() => onPickDataset && onPickDataset(dataset)}>
           {this.props.beforeRow && this.props.beforeRow(dataset)}
           <div className='col'>
             {!!dataset.input.length &&
@@ -102,8 +104,15 @@ class SampleRNNDatasets extends Component {
               />
             }
           </div>
-          <div className='col quiet'>
-            <div>{dataset.isBuilt ? 'cached' : ''}</div>
+          <div className='col quiet hiddenActions'>
+            <div class='subtext'>{dataset.isBuilt ? 'cached' : ''}</div>
+            <div class='actions'>
+              <span class='link' onClick={() => actions.train_task_now(dataset, 1)}>train</span>
+              <span class='link' onClick={() => actions.train_task_now(dataset, 2)}>2x</span>
+              <span class='link' onClick={() => actions.train_task_now(dataset, 4)}>4x</span>
+              <span class='link' onClick={() => actions.train_task_now(dataset, 6)}>6x</span>
+              <span class='link' onClick={() => actions.train_task_now(dataset, 18)}>18x</span>
+            </div>
           </div>
           <div className='col checkpoint'>
             {!!dataset.checkpoints.length &&