fix promise turn

alibaba · May 22, 2020 · 7577026 · 7577026
1 parent 92eac4f
commit 7577026
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/packages/daemon/assets/predict.js b/packages/daemon/assets/predict.js
@@ -24,11 +24,16 @@ const model = modelDefineModule(null, {
 
 function predict(data) {
   const sample = { data, label: null };
+  let future = model;
+
   if (typeof dataProcessModule === 'function') {
-    dataProcessModule(sample, {}, pipeline.dataProcessParams);
+    future = future.then((m) => {
+      dataProcessModule(sample, {}, pipeline.dataProcessParams);
+      return m
+    });
   }
-  return model.then((m) => {
-    return m.predict(sample);
+  return future.then((m) => {
+    m.predict(sample);
   });
 };