add port support in solver

autoai-org · Sep 12, 2018 · ba5aecd · ba5aecd
1 parent 13579cc
commit ba5aecd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/cvpm/server.py b/cvpm/server.py
@@ -88,8 +88,9 @@ def train():
         return json.dumps({"error": "not supported!", "code": "404"}), 404
 
 
-def run_server(solver):
-    port = get_available_port()
+def run_server(solver, port=None):
+    if port is None:
+        port = get_available_port()
     logger.info("Server Running On: " + str(port))
     with server.app_context():
         server.solver = solver

diff --git a/cvpm/solver.py b/cvpm/solver.py
@@ -55,5 +55,5 @@ def infer(self, input, config):
     def train(self, train_x, train_y, **kwargs):
         pass
 
-    def start(self):
-        run_server(self)
+    def start(self, port=None):
+        run_server(self, port)