Adding time series forecasting model deployment

Author: kumar-shivam-ranjan

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):

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,