Fix copy paste error (#649)

(cherry picked from commit a354308)
cortexlabs · Dec 11, 2019 · 8f1c5dd · 8f1c5dd
1 parent 019ce09
commit 8f1c5dd
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pkg/workloads/cortex/onnx_serve/api.py b/pkg/workloads/cortex/onnx_serve/api.py
@@ -309,8 +309,8 @@ def start(args):
             cx_logger().warn("an error occurred while attempting to load classes", exc_info=True)
 
     waitress_kwargs = {}
-    if api["tensorflow"].get("metadata") is not None:
-        for key, value in api["tensorflow"]["metadata"].items():
+    if api["onnx"].get("metadata") is not None:
+        for key, value in api["onnx"]["metadata"].items():
             if key.startswith("waitress_"):
                 waitress_kwargs[key[len("waitress_") :]] = value
 

diff --git a/pkg/workloads/cortex/predictor_serve/api.py b/pkg/workloads/cortex/predictor_serve/api.py
@@ -150,8 +150,8 @@ def start(args):
             cx_logger().warn("an error occurred while attempting to load classes", exc_info=True)
 
     waitress_kwargs = {}
-    if api["tensorflow"].get("metadata") is not None:
-        for key, value in api["tensorflow"]["metadata"].items():
+    if api["predictor"].get("metadata") is not None:
+        for key, value in api["predictor"]["metadata"].items():
             if key.startswith("waitress_"):
                 waitress_kwargs[key[len("waitress_") :]] = value