We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4d88ae8 commit 2488d87Copy full SHA for 2488d87
pkg/cortex/serve/cortex_internal/lib/type/type.py
@@ -59,6 +59,8 @@ def predictor_type_from_api_spec(api_spec: dict) -> PredictorType:
59
"""
60
Get predictor type from API spec.
61
62
- if api_spec["compute"]["inf"] > 0 and api_spec["predictor"]["type"] == TensorFlowPredictorType:
+ if api_spec["compute"]["inf"] > 0 and api_spec["predictor"]["type"] == str(
63
+ TensorFlowPredictorType
64
+ ):
65
return predictor_type_from_string("tensorflow-neuron")
66
return predictor_type_from_string(api_spec["predictor"]["type"])
0 commit comments