1 files changed, 26 insertions, 1 deletions
diff --git a/app/client/modules/pix2wav/pix2wav.actions.js b/app/client/modules/pix2wav/pix2wav.actions.js
index 08f1a97..9d819a3 100644
--- a/app/client/modules/pix2wav/pix2wav.actions.js
+++ b/app/client/modules/pix2wav/pix2wav.actions.js
@@ -15,6 +15,9 @@ export const load_directories = (id) => (dispatch) => {
   const module = pix2wavModule.name
   util.allProgress([
     datasetLoader.load(module),
+    actions.socket.list_directory({ module, dir: 'sequences/pix2wav/' }),
+    actions.socket.list_directory({ module, dir: 'datasets/pix2wav/' }),
+    actions.socket.list_directory({ module, dir: 'checkpoints/pix2wav/' }),
     // actions.socket.list_directory({ module, dir: 'datasets' }),
     // actions.socket.list_directory({ module, dir: 'results' }),
     // actions.socket.list_directory({ module, dir: 'output' }),
@@ -22,7 +25,7 @@ export const load_directories = (id) => (dispatch) => {
   ], (percent, i, n) => {
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
-    const [datasetApiReport] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const [datasetApiReport, sequences, datasets, checkpoints] = res
     const {
       folderLookup,
       fileLookup,
@@ -31,6 +34,24 @@ export const load_directories = (id) => (dispatch) => {
       files,
       unsortedFolder,
     } = datasetApiReport
+
+    const sequenceDirectories = sequences.filter(s => s.dir)
+    sequenceDirectories.forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.isBuilt = true
+    })
+
+    datasets.filter(s => s.dir).forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasDataset = true
+    })
+
+    const checkpointDirectories = checkpoints.filter(s => s.dir)
+    checkpointDirectories.forEach(dir => {
+      const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
+      dataset.hasCheckpoints = true
+    })
+
     dispatch({
       type: types.dataset.load,
       data: {
@@ -39,8 +60,12 @@ export const load_directories = (id) => (dispatch) => {
         fileLookup,
         datasetLookup,
         folders, files,
+        sequences: sequenceDirectories,
+        datasets,
+        checkpoints: checkpointDirectories,
       },
     })
+
     if (id) {
       console.log('folder id', id)
       dispatch({