Skip to content

[AQUA] Time series forecasting model support #1220

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 8 commits into from
Jul 9, 2025
Merged
1 change: 1 addition & 0 deletions ads/aqua/model/constants.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
14 changes: 13 additions & 1 deletion ads/aqua/modeldeployment/deployment.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,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,
Expand Down Expand Up @@ -215,6 +219,14 @@ def create(
freeform_tags=freeform_tags,
defined_tags=defined_tags,
)
task_tag = aqua_model.freeform_tags.get(Tags.TASK, UNKNOWN)
if (
task_tag == 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}
)
return self._create(
aqua_model=aqua_model,
create_deployment_details=create_deployment_details,
Expand Down
Loading