From 8f1c5dd7f610d8fa364645e4b8021e8b61fcd9a4 Mon Sep 17 00:00:00 2001 From: Vishal Bollu Date: Wed, 11 Dec 2019 18:36:36 -0500 Subject: [PATCH] Fix copy paste error (#649) (cherry picked from commit a354308512929584c90592170baf66583dd85bca) --- pkg/workloads/cortex/onnx_serve/api.py | 4 ++-- pkg/workloads/cortex/predictor_serve/api.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pkg/workloads/cortex/onnx_serve/api.py b/pkg/workloads/cortex/onnx_serve/api.py index d54877342e..f89ac15064 100644 --- 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 index f28deb988e..19a29cc94c 100644 --- 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