fix testing

huridocs · Oct 28, 2024 · a2f56bf · a2f56bf
1 parent f5d8231
commit a2f56bf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/app/api/suggestions/routes.ts b/app/api/suggestions/routes.ts
@@ -30,7 +30,6 @@ async function processTrainFunction(
     return;
   }
 
-  await IX.stop();
   const status = await callback(new ObjectId(req.body.extractorId));
   res.json(status);
 }
@@ -137,6 +136,7 @@ export const suggestionsRoutes = (app: Application) => {
     async (req, res, _next) => {
       const IX = new InformationExtraction();
       await processTrainFunction(IX.stopModel, req, res);
+      await IX.stop();
     }
   );