Skip to content

Commit 1bba8e7

Browse files
Adding generic check for task tag
1 parent 49f82b4 commit 1bba8e7

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,9 +218,10 @@ def create(
218218
freeform_tags=freeform_tags,
219219
defined_tags=defined_tags,
220220
)
221+
task_tag = aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN)
221222
if (
222-
aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN)
223-
== ModelTask.TIME_SERIES_FORECASTING
223+
task_tag == ModelTask.TIME_SERIES_FORECASTING
224+
or task_tag.replace("-", "_") == ModelTask.TIME_SERIES_FORECASTING
224225
):
225226
create_deployment_details.env_var.update(
226227
{Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING}

0 commit comments

Comments
 (0)