From 6be0bc13016dbafc1dd4109da51c3b3aba366c3c Mon Sep 17 00:00:00 2001 From: Kumar Ranjan Date: Fri, 4 Jul 2025 10:35:45 +0530 Subject: [PATCH 1/5] Adding time series forecasting model deployment --- ads/aqua/model/constants.py | 1 + ads/aqua/modeldeployment/deployment.py | 12 +++++++++++- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/ads/aqua/model/constants.py b/ads/aqua/model/constants.py index 194245fe4..7daeaa57b 100644 --- a/ads/aqua/model/constants.py +++ b/ads/aqua/model/constants.py @@ -26,6 +26,7 @@ class ModelTask(ExtendedEnum): TEXT_GENERATION = "text-generation" IMAGE_TEXT_TO_TEXT = "image-text-to-text" IMAGE_TO_TEXT = "image-to-text" + TIME_SERIES_FORECASTING = "time-series-forecasting" class FineTuningMetricCategories(ExtendedEnum): diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index f86881b75..8d1465e5b 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -47,7 +47,11 @@ ) from ads.aqua.data import AquaResourceIdentifier from ads.aqua.model import AquaModelApp -from ads.aqua.model.constants import AquaModelMetadataKeys, ModelCustomMetadataFields +from ads.aqua.model.constants import ( + AquaModelMetadataKeys, + ModelCustomMetadataFields, + ModelTask, +) from ads.aqua.model.utils import ( extract_base_model_from_ft, extract_fine_tune_artifacts_path, @@ -214,6 +218,12 @@ def create( freeform_tags=freeform_tags, defined_tags=defined_tags, ) + if aqua_model.freeform_tags.get( + Tags.TASK, ModelTask.TIME_SERIES_FORECASTING + ): + create_deployment_details.env_var.update( + {Tags.TASK.upper(), ModelTask.TIME_SERIES_FORECASTING} + ) return self._create( aqua_model=aqua_model, create_deployment_details=create_deployment_details, From 85bbd20c633ad3c6925d9bdde6321e20238c2944 Mon Sep 17 00:00:00 2001 From: Kumar Ranjan Date: Mon, 7 Jul 2025 13:05:29 +0530 Subject: [PATCH 2/5] Updating dict update func --- ads/aqua/modeldeployment/deployment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index 8d1465e5b..b5f8af485 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -222,7 +222,7 @@ def create( Tags.TASK, ModelTask.TIME_SERIES_FORECASTING ): create_deployment_details.env_var.update( - {Tags.TASK.upper(), ModelTask.TIME_SERIES_FORECASTING} + {Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING} ) return self._create( aqua_model=aqua_model, From 49f82b48b9b2807ca44f252405e9d54c650877a7 Mon Sep 17 00:00:00 2001 From: Kumar Ranjan Date: Mon, 7 Jul 2025 14:35:49 +0530 Subject: [PATCH 3/5] Updating check --- ads/aqua/modeldeployment/deployment.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index b5f8af485..f9a09ab19 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -218,8 +218,9 @@ def create( freeform_tags=freeform_tags, defined_tags=defined_tags, ) - if aqua_model.freeform_tags.get( - Tags.TASK, ModelTask.TIME_SERIES_FORECASTING + if ( + aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN) + == ModelTask.TIME_SERIES_FORECASTING ): create_deployment_details.env_var.update( {Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING} From 1bba8e711b30d6773de53f4573c256ad74a63ef9 Mon Sep 17 00:00:00 2001 From: Kumar Ranjan Date: Tue, 8 Jul 2025 09:45:45 +0530 Subject: [PATCH 4/5] Adding generic check for task tag --- ads/aqua/modeldeployment/deployment.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index f9a09ab19..b7874007a 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -218,9 +218,10 @@ def create( freeform_tags=freeform_tags, defined_tags=defined_tags, ) + task_tag = aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN) if ( - aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN) - == ModelTask.TIME_SERIES_FORECASTING + task_tag == ModelTask.TIME_SERIES_FORECASTING + or task_tag.replace("-", "_") == ModelTask.TIME_SERIES_FORECASTING ): create_deployment_details.env_var.update( {Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING} From 66eb66761e029fbb1de86396d7bbb44cc8bcaafb Mon Sep 17 00:00:00 2001 From: Kumar Ranjan Date: Tue, 8 Jul 2025 13:05:06 +0530 Subject: [PATCH 5/5] Adding generic check for task tag --- ads/aqua/modeldeployment/deployment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ads/aqua/modeldeployment/deployment.py b/ads/aqua/modeldeployment/deployment.py index b7874007a..4f44b0f53 100644 --- a/ads/aqua/modeldeployment/deployment.py +++ b/ads/aqua/modeldeployment/deployment.py @@ -221,7 +221,7 @@ def create( task_tag = aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN) if ( task_tag == ModelTask.TIME_SERIES_FORECASTING - or task_tag.replace("-", "_") == ModelTask.TIME_SERIES_FORECASTING + or task_tag == ModelTask.TIME_SERIES_FORECASTING.replace("-", "_") ): create_deployment_details.env_var.update( {Tags.TASK.upper(): ModelTask.TIME_SERIES_FORECASTING}