Fix typo in model server initialization (#495)

mlrun · Oct 25, 2020 · 6d76aa3 · 6d76aa3
1 parent 0bf0a0b
commit 6d76aa3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mlrun/runtimes/__init__.py b/mlrun/runtimes/__init__.py
@@ -49,7 +49,7 @@ def new_model_server(
     handler=None,
 ):
     if protocol:
-        new_v2_model_server(
+        return new_v2_model_server(
             name,
             model_class,
             models=models,
@@ -61,7 +61,7 @@ def new_model_server(
             canary=canary,
         )
     else:
-        new_v1_model_server(
+        return new_v1_model_server(
             name,
             model_class,
             models=models,