Skip to content

Commit 2488d87

Browse files
committed
Fix TensorFlow Inferentia for LR/MMC
1 parent 4d88ae8 commit 2488d87

File tree

1 file changed

+3
-1
lines changed
  • pkg/cortex/serve/cortex_internal/lib/type

1 file changed

+3
-1
lines changed

pkg/cortex/serve/cortex_internal/lib/type/type.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,8 @@ def predictor_type_from_api_spec(api_spec: dict) -> PredictorType:
5959
"""
6060
Get predictor type from API spec.
6161
"""
62-
if api_spec["compute"]["inf"] > 0 and api_spec["predictor"]["type"] == TensorFlowPredictorType:
62+
if api_spec["compute"]["inf"] > 0 and api_spec["predictor"]["type"] == str(
63+
TensorFlowPredictorType
64+
):
6365
return predictor_type_from_string("tensorflow-neuron")
6466
return predictor_type_from_string(api_spec["predictor"]["type"])

0 commit comments

Comments
 (0)