From 521b024439b202be03447188925869100904b807 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Mon, 4 Jun 2018 19:33:21 +0200
Subject: refactor state

---
 app/client/modules/samplernn/samplernn.datasets.js | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

(limited to 'app/client/modules/samplernn/samplernn.datasets.js')

diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index b4f0c12..ceb8f46 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -84,20 +84,21 @@ class SampleRNNDatasets extends Component {
   }
   renderGroups(){
     const { samplernn, runner, onPickDataset, onPickFile, datasetActions } = this.props
-    const folder = samplernn.folder
+    const { datasetLookup, fileLookup } = samplernn.data
+    const { folder } = samplernn
     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 datasets = folder.datasets.map(name => datasetLookup[name]).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)}>
+        <div key={dataset.name} className='row dataset' onClick={() => onPickDataset && onPickDataset(dataset)}>
           {this.props.beforeRow && this.props.beforeRow(dataset)}
           <div className='col'>
             {!!dataset.input.length &&
               <FileList
-                files={dataset.input}
+                files={dataset.input.map(id => fileLookup[id])}
                 className='input_files'
                 fileListClassName=''
                 rowClassName='input_file'
@@ -122,7 +123,7 @@ class SampleRNNDatasets extends Component {
           <div className='col'>
             {!!dataset.output.length &&
               <FileList
-                files={dataset.output}
+                files={dataset.output.map(id => fileLookup[id])}
                 orderBy='epoch desc'
                 fields={'name date epoch size'}
                 onPickFile={onPickFile}
-- 
cgit v1.2.3-70-g09d2