1 files changed, 6 insertions, 5 deletions
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}