more tasklist stuff

1 files changed, 13 insertions, 7 deletions

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 8fec652..89184d2 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -23,10 +23,12 @@ export const fetch_task = (url, file_id, dataset) => dispatch => {
 }
 
 export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
+  dataset = dataset.name || dataset
+  if (dataset === 'PLACEHOLDER') return
   const task = {
     module: module.name,
     activity: 'train',
-    dataset: dataset.name || dataset,
+    dataset,
     epoch: 0,
     epochs: epochs,
     opt: {
@@ -58,6 +60,7 @@ export const live_task = (sequence, checkpoint, opt) => dispatch => {
 }
 
 export const augment_task = (dataset, opt) => dispatch => {
+  if (dataset === 'PLACEHOLDER') return
   const task = {
     module: module.name,
     activity: 'augment',
@@ -72,14 +75,17 @@ export const augment_task = (dataset, opt) => dispatch => {
   return actions.queue.add_task(task)
 }
 
-export const clear_recursive_task = (dataset) => dispatch => {
+export const create_dataset_task = (opt) => dispatch => {
   const task = {
     module: module.name,
-    activity: 'clear_recursive',
-    dataset,
+    activity: 'create_dataset',
+    dataset: opt.title,
+    folder_id: opt.folder_id,
+    opt: {
+      ...opt,
+    }
   }
   console.log(task)
-  console.log('add clear recursive task')
+  console.log('add create_dataset task')
   return actions.queue.add_task(task)
-}
-
+}
\ No newline at end of file