fix position of disable route

camicroscope · Oct 15, 2021 · e60751f · e60751f
1 parent 3acb140
commit e60751f
Showing 1 changed file with 17 additions and 17 deletions.
diff --git a/caracal.js b/caracal.js
@@ -29,23 +29,6 @@ if (!DISABLE_TF) {
   const Model = require('./handlers/modelTrainer.js');
 }
 
-if (!DISABLE_TF) {
-  HANDLERS["getDataset"] = DataSet.getDataset;
-  HANDLERS["trainModel"] = Model.trainModel;
-  HANDLERS["deleteDataset"] = DataSet.deleteData;
-  HANDLERS["sendTrainedModel"] = Model.sendTrainedModel;
-} else {
-  function disabledRoute() {
-    return function(req, res) {
-      res.status(500).send('{"err":"This TF route is disabled"}');
-    };
-  }
-  HANDLERS["getDataset"] = disabledRoute;
-  HANDLERS["trainModel"] = disabledRoute;
-  HANDLERS["deleteDataset"] = disabledRoute;
-  HANDLERS["sendTrainedModel"] = disabledRoute;
-}
-
 const {connector} = require("./service/database/connector");
 
 var WORKERS = process.env.NUM_THREADS || 4;
@@ -133,6 +116,23 @@ var HANDLERS = {
   },
 };
 
+if (!DISABLE_TF) {
+  HANDLERS["getDataset"] = DataSet.getDataset;
+  HANDLERS["trainModel"] = Model.trainModel;
+  HANDLERS["deleteDataset"] = DataSet.deleteData;
+  HANDLERS["sendTrainedModel"] = Model.sendTrainedModel;
+} else {
+  function disabledRoute() {
+    return function(req, res) {
+      res.status(500).send('{"err":"This TF route is disabled"}');
+    };
+  }
+  HANDLERS["getDataset"] = disabledRoute;
+  HANDLERS["trainModel"] = disabledRoute;
+  HANDLERS["deleteDataset"] = disabledRoute;
+  HANDLERS["sendTrainedModel"] = disabledRoute;
+}
+
 // register configurable services
 // TODO verify all
 for (let i in routeConfig) {