1 files changed, 6 insertions, 12 deletions
diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index 92c0ff4..bd51f2b 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -53,34 +53,28 @@ export const live_task = (sequence, checkpoint, opt) => dispatch => {
   return actions.queue.add_task(task)
 }
 
-export const augment_task = (opt) => dispatch => {
+export const augment_task = (dataset, opt) => dispatch => {
   const task = {
     module: module.name,
     activity: 'augment',
-    dataset: sequence,
-    checkpoint,
+    dataset,
     opt: {
       ...opt,
-      poll_delay: 0.01,
     }
   }
   console.log(task)
-  console.log('add live task')
+  console.log('add augment task')
   return actions.queue.add_task(task)
 }
 
-export const clear_recursive_task = (opt) => dispatch => {
+export const clear_recursive_task = (dataset) => dispatch => {
   const task = {
     module: module.name,
     activity: 'clear_recursive',
-    dataset: sequence,
-    checkpoint,
-    opt: {
-      ...opt,
-    }
+    dataset,
   }
   console.log(task)
-  console.log('add live task')
+  console.log('add clear recursive task')
   return actions.queue.add_task(task)
 }