1 files changed, 5 insertions, 4 deletions
diff --git a/app/relay/modules/samplernn.js b/app/relay/modules/samplernn.js
index 2dacaa7..73d0a5f 100644
--- a/app/relay/modules/samplernn.js
+++ b/app/relay/modules/samplernn.js
@@ -13,10 +13,11 @@ const fetch = {
   listen: (task, line, i) => {
     // here i need to bridge again... get the filename that comes back from youtube-dl
     // and tell the cortex that URL -> fn and add the filename!
-    if ( line.match(/^got fn, /) ) {
-      let dataset = line.split(' => ')[1].trim()
-      task.dataset = dataset
-      return { type: 'progress', action: 'set_dataset', task, }
+    if ( line.match(/^youtube-dl got fn, /) ) {
+      let filename = line.split(' => ')[1].trim()
+      task.dataset = filename.split('.')[0]
+      task.opt.filename = filename
+      return { type: 'progress', action: 'resolve_dataset', task, }
     }
     return null
   }