From a4f6d0e9afa4993437e3031aaa4771f268ce4d99 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Wed, 5 Sep 2018 13:28:55 +0200
Subject: app

---
 app/client/modules/pix2pixhd/pix2pixhd.tasks.js | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

(limited to 'app/client/modules/pix2pixhd/pix2pixhd.tasks.js')

diff --git a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
index bd51f2b..412f066 100644
--- a/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
+++ b/app/client/modules/pix2pixhd/pix2pixhd.tasks.js
@@ -26,13 +26,17 @@ export const train_task = (dataset, folder_id, epochs=1) => dispatch => {
   const task = {
     module: module.name,
     activity: 'train',
-    dataset: dataset.name,
+    dataset: dataset.name || dataset,
     epoch: 0,
     epochs: epochs,
     opt: {
       folder_id: folder_id,
     }
   }
+  if (!task.dataset) {
+    console.error("train task: no dataset specified")
+    return
+  }
   console.log(task)
   return actions.queue.add_task(task)
 }
-- 
cgit v1.2.3-70-g09d2