Skip to content

Commit 49f82b4

Browse files
Updating check
1 parent 85bbd20 commit 49f82b4

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ads/aqua/modeldeployment/deployment.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,8 +218,9 @@ def create(
218218
freeform_tags=freeform_tags,
219219
defined_tags=defined_tags,
220220
)
221-
if aqua_model.freeform_tags.get(
222-
Tags.TASK, ModelTask.TIME_SERIES_FORECASTING
221+
if (
222+
aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN)
223+
== ModelTask.TIME_SERIES_FORECASTING
223224
):
224225
create_deployment_details.env_var.update(
225226
{Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING}

0 commit comments

Comments
 (0)