1 files changed, 10 insertions, 1 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
+      },
     })
   })
 }