2 files changed, 13 insertions, 4 deletions
diff --git a/app/client/modules/samplernn/samplernn.actions.js b/app/client/modules/samplernn/samplernn.actions.js
index 8338c82..55bd9f3 100644
--- a/app/client/modules/samplernn/samplernn.actions.js
+++ b/app/client/modules/samplernn/samplernn.actions.js
@@ -17,11 +17,20 @@ export const load_directories = () => (dispatch) => {
   ]).then(res => {
     console.log(res)
     const [folders, files, tasks, results, datasets, report] = res
+    console.log(datasets)
+    const builtDatasets = datasets.filter(s => s.dir)
+    const flatDatasets = datasets.filter(s => s.name.match(/(wav|aiff?|flac|mp3)$/) && !s.dir)
+    const checkpoints = results.filter(s => s.dir)
     console.log(folders.length, files.length, tasks.length, results.length, datasets.length)
     console.log(report)
     dispatch({
       type: types.samplernn.init,
-      data: { folders, files, results, datasets, report },
+      data: {
+        folders, files,
+        checkpoints,
+        builtDatasets, flatDatasets,
+        report
+      },
     })
   })
 }
diff --git a/app/client/modules/samplernn/samplernn.datasets.js b/app/client/modules/samplernn/samplernn.datasets.js
index f358170..58e6df4 100644
--- a/app/client/modules/samplernn/samplernn.datasets.js
+++ b/app/client/modules/samplernn/samplernn.datasets.js
@@ -90,11 +90,11 @@ class SampleRNNDatasets extends Component {
         />
         <FileList
           title='Datasets'
-          files={samplernn.data.datasets}
+          files={samplernn.data.builtDatasets}
         />
         <FileList
-          title='Results'
-          files={samplernn.data.results}
+          title='Checkpoints'
+          files={samplernn.data.checkpoints}
         />
       </div>
     )