action divs

1 files changed, 9 insertions, 12 deletions

diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index d29cf6a..b4f0c12 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -72,9 +72,9 @@ class SampleRNNDatasets extends Component {
     if (!folder || !folder.name) return
     return (
       <div class='rows params datasets'>
-        <div class='row dataset'>
+        <div class='row row-heading dataset'>
           <div class='col'>input</div>
-          <div class='col'></div>
+          <div class='col'>status</div>
           <div class='col'>checkpoint</div>
           <div class='col'>output</div>
         </div>
@@ -83,11 +83,14 @@ class SampleRNNDatasets extends Component {
     )
   }
   renderGroups(){
-    const { samplernn, onPickDataset, onPickFile, actions } = this.props
+    const { samplernn, runner, onPickDataset, onPickFile, datasetActions } = this.props
     const folder = samplernn.folder
     const { mapFn, sortFn } = util.sort.orderByFn('date desc')
+    const moduleOnGPU = runner.gpu.task.module === samplernnModule.name
     const datasets = folder.datasets.map(mapFn).sort(sortFn).map(pair => {
       const dataset = pair[1]
+      const isProcessing = moduleOnGPU && runner.gpu.task.dataset === dataset
+      const status = isProcessing ? util.gerund(runner.gpu.task.activity) : ''
       return (
         <div className='row dataset' onClick={() => onPickDataset && onPickDataset(dataset)}>
           {this.props.beforeRow && this.props.beforeRow(dataset)}
@@ -103,15 +106,9 @@ class SampleRNNDatasets extends Component {
               />
             }
           </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 className={'col quiet ' + (dataset.isBuilt ? 'built' : 'not_built')}>
+            {status}
+            {this.props.datasetActions && this.props.datasetActions(dataset)}
           </div>
           <div className='col checkpoint'>
             {!!dataset.checkpoints.length &&