Skip to content

Commit 60b803f

Browse files
committed
Update external model check in api.py
1 parent f620125 commit 60b803f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

pkg/workloads/tf_api/api.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -376,7 +376,7 @@ def start(args):
376376
local_cache["api"] = api
377377
local_cache["ctx"] = ctx
378378

379-
if ctx.environment is not None:
379+
if api.get("external_model") is None:
380380
model = ctx.models[api["model_name"]]
381381
estimator = ctx.estimators[model["estimator"]]
382382

0 commit comments

Comments
 (0)