mandatory results folder

1 files changed, 7 insertions, 4 deletions

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.actions.js b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
index caf6b45..917b375 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.actions.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.actions.js
@@ -31,15 +31,16 @@ export const load_directories = (id) => (dispatch) => {
       folders,
       files,
       unsortedFolder,
+      resultsFolder,
     } = datasetApiReport
     // console.log(datasetUsage)
 
     const sequenceDirectories = sequences.filter(s => s.dir)
-    console.log(sequenceDirectories)
+    // console.log(sequenceDirectories)
     sequenceDirectories.forEach(dir => {
       const dataset = datasetLoader.getDataset(module, datasetLookup, dir.name)
       dataset.isBuilt = true
-      console.log(dir.name, dataset)
+      // console.log(dir.name, dataset)
     })
 
     datasets.filter(s => s.dir).forEach(dir => {
@@ -119,6 +120,7 @@ export const load_directories = (id) => (dispatch) => {
         sequences: sequenceDirectories,
         datasets,
         checkpoints: checkpointDirectories,
+        resultsFolder,
       },
     })
     if (id) {
@@ -137,9 +139,9 @@ export const load_directories = (id) => (dispatch) => {
 }
 
 export const load_results = (id) => (dispatch) => {
-  console.log('hi')
   const module = pix2pixhdModule.name
   util.allProgress([
+    actions.folder.index({ name: 'results' }),
     actions.file.index({ module, generated: 1 }),
     actions.socket.list_sequences({ module, dir: 'results' }),
     actions.socket.list_directory({ module, dir: 'renders' }),
@@ -147,11 +149,12 @@ export const load_results = (id) => (dispatch) => {
     console.log('pix2pixhd load progress', i, n)
     dispatch({ type: types.app.load_progress, progress: { i, n }})
   }).then(res => {
-    const [files, results, renders] = res //, datasets, results, output, datasetUsage, lossReport] = res
+    const [folders, files, results, renders] = res //, datasets, results, output, datasetUsage, lossReport] = res
     console.log(files, results, renders)
     dispatch({
       type: types.pix2pixhd.load_results,
       results: {
+        folder: folders[0],
         files,
         results,
         renders,