oracle-ads 2.12.10rc0__py3-none-any.whl → 2.13.0__py3-none-any.whl

ads/opctl/operator/common/backend_factory.py

@@ -1,7 +1,6 @@
 #!/usr/bin/env python
-# -*- coding: utf-8 -*--
 
-# Copyright (c) 2023, 2024 Oracle and/or its affiliates.
+# Copyright (c) 2023, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 """
@@ -12,11 +11,6 @@ The factory validates the backend type and runtime type before creating the back
 from typing import Dict, List, Tuple, Union
 
 import yaml
-from ads.opctl.operator.common.utils import print_traceback
-
-from ads.opctl.backend.marketplace.local_marketplace import (
-    LocalMarketplaceOperatorBackend,
-)
 
 from ads.opctl import logger
 from ads.opctl.backend.ads_dataflow import DataFlowOperatorBackend
@@ -25,6 +19,9 @@ from ads.opctl.backend.base import Backend
 from ads.opctl.backend.local import (
     LocalOperatorBackend,
 )
+from ads.opctl.backend.marketplace.local_marketplace import (
+    LocalMarketplaceOperatorBackend,
+)
 from ads.opctl.config.base import ConfigProcessor
 from ads.opctl.config.merger import ConfigMerger
 from ads.opctl.constants import (
@@ -34,9 +31,10 @@ from ads.opctl.constants import (
     RESOURCE_TYPE,
     RUNTIME_TYPE,
 )
-from ads.opctl.operator.common.const import PACK_TYPE, OPERATOR_BACKEND_SECTION_NAME
+from ads.opctl.operator.common.const import OPERATOR_BACKEND_SECTION_NAME, PACK_TYPE
 from ads.opctl.operator.common.dictionary_merger import DictionaryMerger
 from ads.opctl.operator.common.operator_loader import OperatorInfo, OperatorLoader
+from ads.opctl.operator.common.utils import print_traceback
 
 
 class BackendFactory:
@@ -46,57 +44,57 @@ class BackendFactory:
     """
 
     BACKENDS = (
-        BACKEND_NAME.JOB.value,
-        BACKEND_NAME.DATAFLOW.value,
-        BACKEND_NAME.MARKETPLACE.value,
+        BACKEND_NAME.JOB,
+        BACKEND_NAME.DATAFLOW,
+        BACKEND_NAME.MARKETPLACE,
     )
 
     LOCAL_BACKENDS = (
-        BACKEND_NAME.OPERATOR_LOCAL.value,
-        BACKEND_NAME.LOCAL.value,
+        BACKEND_NAME.OPERATOR_LOCAL,
+        BACKEND_NAME.LOCAL,
     )
 
     BACKEND_RUNTIME_MAP = {
-        BACKEND_NAME.JOB.value.lower(): {
-            RUNTIME_TYPE.PYTHON.value.lower(): (
-                BACKEND_NAME.JOB.value.lower(),
-                RUNTIME_TYPE.PYTHON.value.lower(),
+        BACKEND_NAME.JOB.lower(): {
+            RUNTIME_TYPE.PYTHON.lower(): (
+                BACKEND_NAME.JOB.lower(),
+                RUNTIME_TYPE.PYTHON.lower(),
             ),
-            RUNTIME_TYPE.CONTAINER.value.lower(): (
-                BACKEND_NAME.JOB.value.lower(),
-                RUNTIME_TYPE.CONTAINER.value.lower(),
+            RUNTIME_TYPE.CONTAINER.lower(): (
+                BACKEND_NAME.JOB.lower(),
+                RUNTIME_TYPE.CONTAINER.lower(),
             ),
         },
-        BACKEND_NAME.DATAFLOW.value.lower(): {
-            RUNTIME_TYPE.DATAFLOW.value.lower(): (
-                BACKEND_NAME.DATAFLOW.value.lower(),
-                RUNTIME_TYPE.DATAFLOW.value.lower(),
+        BACKEND_NAME.DATAFLOW.lower(): {
+            RUNTIME_TYPE.DATAFLOW.lower(): (
+                BACKEND_NAME.DATAFLOW.lower(),
+                RUNTIME_TYPE.DATAFLOW.lower(),
             )
         },
-        BACKEND_NAME.OPERATOR_LOCAL.value.lower(): {
-            RUNTIME_TYPE.PYTHON.value.lower(): (
-                BACKEND_NAME.OPERATOR_LOCAL.value.lower(),
-                RUNTIME_TYPE.PYTHON.value.lower(),
+        BACKEND_NAME.OPERATOR_LOCAL.lower(): {
+            RUNTIME_TYPE.PYTHON.lower(): (
+                BACKEND_NAME.OPERATOR_LOCAL.lower(),
+                RUNTIME_TYPE.PYTHON.lower(),
             ),
-            RUNTIME_TYPE.CONTAINER.value.lower(): (
-                BACKEND_NAME.OPERATOR_LOCAL.value.lower(),
-                RUNTIME_TYPE.CONTAINER.value.lower(),
+            RUNTIME_TYPE.CONTAINER.lower(): (
+                BACKEND_NAME.OPERATOR_LOCAL.lower(),
+                RUNTIME_TYPE.CONTAINER.lower(),
             ),
         },
-        BACKEND_NAME.MARKETPLACE.value.lower(): {
-            RUNTIME_TYPE.PYTHON.value.lower(): (
-                BACKEND_NAME.MARKETPLACE.value.lower(),
-                RUNTIME_TYPE.PYTHON.value.lower(),
+        BACKEND_NAME.MARKETPLACE.lower(): {
+            RUNTIME_TYPE.PYTHON.lower(): (
+                BACKEND_NAME.MARKETPLACE.lower(),
+                RUNTIME_TYPE.PYTHON.lower(),
             )
         },
     }
 
     BACKEND_MAP = {
-        BACKEND_NAME.JOB.value.lower(): MLJobOperatorBackend,
-        BACKEND_NAME.DATAFLOW.value.lower(): DataFlowOperatorBackend,
-        BACKEND_NAME.OPERATOR_LOCAL.value.lower(): LocalOperatorBackend,
-        BACKEND_NAME.LOCAL.value.lower(): LocalOperatorBackend,
-        BACKEND_NAME.MARKETPLACE.value.lower(): LocalMarketplaceOperatorBackend,
+        BACKEND_NAME.JOB.lower(): MLJobOperatorBackend,
+        BACKEND_NAME.DATAFLOW.lower(): DataFlowOperatorBackend,
+        BACKEND_NAME.OPERATOR_LOCAL.lower(): LocalOperatorBackend,
+        BACKEND_NAME.LOCAL.lower(): LocalOperatorBackend,
+        BACKEND_NAME.MARKETPLACE.lower(): LocalMarketplaceOperatorBackend,
     }
 
     @classmethod
@@ -135,15 +133,15 @@ class BackendFactory:
         # validation
         if not operator_type:
             raise RuntimeError(
-                f"The `type` attribute must be specified in the operator's config."
+                "The `type` attribute must be specified in the operator's config."
             )
 
         if not backend and not config.config.get(OPERATOR_BACKEND_SECTION_NAME):
             logger.info(
-                f"Backend config is not provided, the {BACKEND_NAME.LOCAL.value} "
+                f"Backend config is not provided, the {BACKEND_NAME.LOCAL} "
                 "will be used by default. "
             )
-            backend = BACKEND_NAME.LOCAL.value
+            backend = BACKEND_NAME.LOCAL
         elif not backend:
             backend = config.config.get(OPERATOR_BACKEND_SECTION_NAME)
 
@@ -164,8 +162,8 @@ class BackendFactory:
             backend = {"kind": backend_kind}
 
         backend_kind = (
-            BACKEND_NAME.OPERATOR_LOCAL.value
-            if backend.get("kind").lower() == BACKEND_NAME.LOCAL.value
+            BACKEND_NAME.OPERATOR_LOCAL
+            if backend.get("kind").lower() == BACKEND_NAME.LOCAL
             else backend.get("kind").lower()
         )
         backend["kind"] = backend_kind
@@ -174,11 +172,11 @@ class BackendFactory:
         # This is necessary, because Jobs and DataFlow have similar kind,
         # The only difference would be in the infrastructure kind.
         # This is a temporary solution, the logic needs to be placed in the ConfigMerger instead.
-        if backend_kind == BACKEND_NAME.JOB.value:
+        if backend_kind == BACKEND_NAME.JOB:
             if (backend.get("spec", {}) or {}).get("infrastructure", {}).get(
                 "type", ""
-            ).lower() == BACKEND_NAME.DATAFLOW.value:
-                backend_kind = BACKEND_NAME.DATAFLOW.value
+            ).lower() == BACKEND_NAME.DATAFLOW:
+                backend_kind = BACKEND_NAME.DATAFLOW
 
         runtime_type = runtime_type or (
             backend.get("type")
@@ -247,17 +245,17 @@ class BackendFactory:
             If the backend type is not supported.
         """
         supported_backends = supported_backends or (cls.BACKENDS + cls.LOCAL_BACKENDS)
-        backend = (backend or BACKEND_NAME.OPERATOR_LOCAL.value).lower()
+        backend = (backend or BACKEND_NAME.OPERATOR_LOCAL).lower()
         backend_kind, runtime_type = backend, None
 
-        if backend.lower() != BACKEND_NAME.OPERATOR_LOCAL.value and "." in backend:
+        if backend.lower() != BACKEND_NAME.OPERATOR_LOCAL and "." in backend:
             backend_kind, runtime_type = backend.split(".")
         else:
             backend_kind = backend
 
         backend_kind = (
-            BACKEND_NAME.OPERATOR_LOCAL.value
-            if backend_kind == BACKEND_NAME.LOCAL.value
+            BACKEND_NAME.OPERATOR_LOCAL
+            if backend_kind == BACKEND_NAME.LOCAL
             else backend_kind
         )
 
@@ -357,7 +355,7 @@ class BackendFactory:
 
         # generate supported backend specifications templates YAML
         RUNTIME_TYPE_MAP = {
-            RESOURCE_TYPE.JOB.value: [
+            RESOURCE_TYPE.JOB: [
                 {
                     RUNTIME_TYPE.PYTHON: {
                         "conda_slug": operator_info.conda
@@ -373,7 +371,7 @@ class BackendFactory:
                     }
                 },
             ],
-            RESOURCE_TYPE.DATAFLOW.value: [
+            RESOURCE_TYPE.DATAFLOW: [
                 {
                     RUNTIME_TYPE.DATAFLOW: {
                         "conda_slug": operator_info.conda_prefix,
@@ -381,7 +379,7 @@ class BackendFactory:
                     }
                 }
             ],
-            BACKEND_NAME.OPERATOR_LOCAL.value: [
+            BACKEND_NAME.OPERATOR_LOCAL: [
                 {
                     RUNTIME_TYPE.CONTAINER: {
                         "kind": "operator",
@@ -397,7 +395,7 @@ class BackendFactory:
                     }
                 },
             ],
-            BACKEND_NAME.MARKETPLACE.value: [
+            BACKEND_NAME.MARKETPLACE: [
                 {
                     RUNTIME_TYPE.PYTHON: {
                         "kind": "marketplace",
@@ -445,11 +443,9 @@ class BackendFactory:
                     )
 
                     # generate YAML specification template
-                    result[
-                        (resource_type.lower(), runtime_type.value.lower())
-                    ] = yaml.load(
+                    result[(resource_type.lower(), runtime_type.lower())] = yaml.load(
                         _BackendFactory(p.config).backend.init(
-                            runtime_type=runtime_type.value,
+                            runtime_type=runtime_type,
                             **{**kwargs, **runtime_kwargs},
                         ),
                         Loader=yaml.FullLoader,

ads/opctl/operator/common/const.py

@@ -1,10 +1,9 @@
 #!/usr/bin/env python
-# -*- coding: utf-8 -*--
 
-# Copyright (c) 2023 Oracle and/or its affiliates.
+# Copyright (c) 2023, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
-from ads.common.extended_enum import ExtendedEnumMeta
+from ads.common.extended_enum import ExtendedEnum
 
 # Env variable representing the operator input arguments.
 # This variable is used when operator run on the OCI resources.
@@ -17,11 +16,12 @@ OPERATOR_BASE_DOCKER_GPU_FILE = "Dockerfile.gpu"
 
 OPERATOR_BACKEND_SECTION_NAME = "backend"
 
-class PACK_TYPE(str, metaclass=ExtendedEnumMeta):
+
+class PACK_TYPE(ExtendedEnum):
     SERVICE = "service"
     CUSTOM = "published"
 
 
-class ARCH_TYPE(str, metaclass=ExtendedEnumMeta):
+class ARCH_TYPE(ExtendedEnum):
     CPU = "cpu"
     GPU = "gpu"

ads/opctl/operator/lowcode/anomaly/const.py

@@ -1,14 +1,13 @@
 #!/usr/bin/env python
-# -*- coding: utf-8 -*--
 
-# Copyright (c) 2023 Oracle and/or its affiliates.
+# Copyright (c) 2023, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
-from ads.common.extended_enum import ExtendedEnumMeta
+from ads.common.extended_enum import ExtendedEnum
 from ads.opctl.operator.lowcode.common.const import DataColumns
 
 
-class SupportedModels(str, metaclass=ExtendedEnumMeta):
+class SupportedModels(ExtendedEnum):
     """Supported anomaly models."""
 
     AutoTS = "autots"
@@ -38,7 +37,7 @@ class SupportedModels(str, metaclass=ExtendedEnumMeta):
     BOCPD = "bocpd"
 
 
-class NonTimeADSupportedModels(str, metaclass=ExtendedEnumMeta):
+class NonTimeADSupportedModels(ExtendedEnum):
     """Supported non time-based anomaly detection models."""
 
     OneClassSVM = "oneclasssvm"
@@ -48,7 +47,7 @@ class NonTimeADSupportedModels(str, metaclass=ExtendedEnumMeta):
     # DBScan = "dbscan"
 
 
-class TODSSubModels(str, metaclass=ExtendedEnumMeta):
+class TODSSubModels(ExtendedEnum):
     """Supported TODS sub models."""
 
     OCSVM = "ocsvm"
@@ -78,7 +77,7 @@ TODS_MODEL_MAP = {
 }
 
 
-class MerlionADModels(str, metaclass=ExtendedEnumMeta):
+class MerlionADModels(ExtendedEnum):
     """Supported Merlion AD sub models."""
 
     # point anomaly
@@ -126,7 +125,7 @@ MERLIONAD_MODEL_MAP = {
 }
 
 
-class SupportedMetrics(str, metaclass=ExtendedEnumMeta):
+class SupportedMetrics(ExtendedEnum):
     UNSUPERVISED_UNIFY95 = "unsupervised_unify95"
     UNSUPERVISED_UNIFY95_LOG_LOSS = "unsupervised_unify95_log_loss"
     UNSUPERVISED_N1_EXPERTS = "unsupervised_n-1_experts"
@@ -158,7 +157,7 @@ class SupportedMetrics(str, metaclass=ExtendedEnumMeta):
     ELAPSED_TIME = "Elapsed Time"
 
 
-class OutputColumns(str, metaclass=ExtendedEnumMeta):
+class OutputColumns(ExtendedEnum):
     ANOMALY_COL = "anomaly"
     SCORE_COL = "score"
     Series = DataColumns.Series

ads/opctl/operator/lowcode/common/transformations.py

@@ -15,6 +15,7 @@ from ads.opctl.operator.lowcode.common.errors import (
     InvalidParameterError,
 )
 from ads.opctl.operator.lowcode.common.utils import merge_category_columns
+from ads.opctl.operator.lowcode.forecast.operator_config import ForecastOperatorSpec
 
 
 class Transformations(ABC):
@@ -34,6 +35,7 @@ class Transformations(ABC):
         self.dataset_info = dataset_info
         self.target_category_columns = dataset_info.target_category_columns
         self.target_column_name = dataset_info.target_column
+        self.raw_column_names = None
         self.dt_column_name = (
             dataset_info.datetime_column.name if dataset_info.datetime_column else None
         )
@@ -60,7 +62,8 @@ class Transformations(ABC):
 
         """
         clean_df = self._remove_trailing_whitespace(data)
-        # clean_df = self._normalize_column_names(clean_df)
+        if isinstance(self.dataset_info, ForecastOperatorSpec):
+            clean_df = self._clean_column_names(clean_df)
         if self.name == "historical_data":
             self._check_historical_dataset(clean_df)
         clean_df = self._set_series_id_column(clean_df)
@@ -98,8 +101,36 @@ class Transformations(ABC):
     def _remove_trailing_whitespace(self, df):
         return df.apply(lambda x: x.str.strip() if x.dtype == "object" else x)
 
-    # def _normalize_column_names(self, df):
-    #     return df.rename(columns=lambda x: re.sub("[^A-Za-z0-9_]+", "", x))
+    def _clean_column_names(self, df):
+        """
+        Remove all whitespaces from column names in a DataFrame and store the original names.
+
+        Parameters:
+        df (pd.DataFrame): The DataFrame whose column names need to be cleaned.
+
+        Returns:
+        pd.DataFrame: The DataFrame with cleaned column names.
+        """
+
+        self.raw_column_names = {
+            col: col.replace(" ", "") for col in df.columns if " " in col
+        }
+        df.columns = [self.raw_column_names.get(col, col) for col in df.columns]
+
+        if self.target_column_name:
+            self.target_column_name = self.raw_column_names.get(
+                self.target_column_name, self.target_column_name
+            )
+        self.dt_column_name = self.raw_column_names.get(
+            self.dt_column_name, self.dt_column_name
+        )
+
+        if self.target_category_columns:
+            self.target_category_columns = [
+                self.raw_column_names.get(col, col)
+                for col in self.target_category_columns
+            ]
+        return df
 
     def _set_series_id_column(self, df):
         self._target_category_columns_map = {}
@@ -233,6 +264,10 @@ class Transformations(ABC):
         expected_names = [self.target_column_name, self.dt_column_name] + (
             self.target_category_columns if self.target_category_columns else []
         )
+
+        if self.raw_column_names:
+            expected_names.extend(list(self.raw_column_names.values()))
+
         if set(df.columns) != set(expected_names):
             raise DataMismatchError(
                 f"Expected {self.name} to have columns: {expected_names}, but instead found column names: {df.columns}. Is the {self.name} path correct?"

ads/opctl/operator/lowcode/common/utils.py

@@ -12,6 +12,7 @@ from typing import List, Union
 
 import fsspec
 import oracledb
+import json
 import pandas as pd
 
 from ads.common.object_storage_details import ObjectStorageDetails
@@ -125,7 +126,7 @@ def load_data(data_spec, storage_options=None, **kwargs):
     return data
 
 
-def write_data(data, filename, format, storage_options, index=False, **kwargs):
+def write_data(data, filename, format, storage_options=None, index=False, **kwargs):
     disable_print()
     if not format:
         _, format = os.path.splitext(filename)
@@ -141,6 +142,15 @@ def write_data(data, filename, format, storage_options, index=False, **kwargs):
     )
 
 
+def write_simple_json(data, path):
+    if ObjectStorageDetails.is_oci_path(path):
+        storage_options = default_signer()
+    else:
+        storage_options = {}
+    with fsspec.open(path, mode="w", **storage_options) as f:
+        json.dump(data, f, indent=4)
+
+
 def merge_category_columns(data, target_category_columns):
     result = data.apply(
         lambda x: "__".join([str(x[col]) for col in target_category_columns]), axis=1

ads/opctl/operator/lowcode/feature_store_marketplace/operator_utils.py

@@ -1,71 +1,66 @@
 #!/usr/bin/env python
-# -*- coding: utf-8 -*--
 
-# Copyright (c) 2024 Oracle and/or its affiliates.
+# Copyright (c) 2024, 2025 Oracle and/or its affiliates.
 # Licensed under the Universal Permissive License v 1.0 as shown at https://oss.oracle.com/licenses/upl/
 
 import ast
 import base64
-from typing import Optional, List, Dict
+from typing import TYPE_CHECKING, Dict, List, Optional
 
 import oci
 import requests
-from typing import TYPE_CHECKING
 
 try:
     from kubernetes.client import (
-        V1ServiceStatus,
-        V1Service,
-        V1LoadBalancerStatus,
         V1LoadBalancerIngress,
+        V1LoadBalancerStatus,
+        V1Service,
+        V1ServiceStatus,
     )
 except ImportError:
     if TYPE_CHECKING:
         from kubernetes.client import (
-            V1ServiceStatus,
-            V1Service,
-            V1LoadBalancerStatus,
             V1LoadBalancerIngress,
+            V1LoadBalancerStatus,
+            V1Service,
+            V1ServiceStatus,
         )
 
-from oci.resource_manager.models import StackSummary, AssociatedResourceSummary
-
-from ads.opctl.operator.lowcode.feature_store_marketplace.models.apigw_config import (
-    APIGatewayConfig,
-)
+import click
+from oci.resource_manager.models import AssociatedResourceSummary, StackSummary
 
-from ads.common.oci_client import OCIClientFactory
-from ads.opctl.operator.lowcode.feature_store_marketplace.const import (
-    LISTING_ID,
-    APIGW_STACK_NAME,
-    STACK_URL,
-    NLB_RULES_ADDRESS,
-    NODES_RULES_ADDRESS,
-)
 from ads import logger
-import click
+from ads.common import auth as authutil
+from ads.common.oci_client import OCIClientFactory
 from ads.opctl import logger
-
 from ads.opctl.backend.marketplace.marketplace_utils import (
     Color,
     print_heading,
     print_ticker,
 )
-from ads.opctl.operator.lowcode.feature_store_marketplace.models.mysql_config import (
-    MySqlConfig,
+from ads.opctl.operator.lowcode.feature_store_marketplace.const import (
+    APIGW_STACK_NAME,
+    LISTING_ID,
+    NLB_RULES_ADDRESS,
+    NODES_RULES_ADDRESS,
+    STACK_URL,
+)
+from ads.opctl.operator.lowcode.feature_store_marketplace.models.apigw_config import (
+    APIGatewayConfig,
 )
-
 from ads.opctl.operator.lowcode.feature_store_marketplace.models.db_config import (
     DBConfig,
 )
-from ads.common import auth as authutil
+from ads.opctl.operator.lowcode.feature_store_marketplace.models.mysql_config import (
+    MySqlConfig,
+)
 
 
 def get_db_details() -> DBConfig:
     jdbc_url = "jdbc:mysql://{}/{}?createDatabaseIfNotExist=true"
     mysql_db_config = MySqlConfig()
     print_heading(
-        f"MySQL database configuration",
+        "MySQL database configuration",
         colors=[Color.BOLD, Color.BLUE],
         prefix_newline_count=2,
     )
@@ -76,12 +71,12 @@ def get_db_details() -> DBConfig:
             "Is password provided as plain-text or via a Vault secret?\n"
             "(https://docs.oracle.com/en-us/iaas/Content/KeyManagement/Concepts/keyoverview.htm)",
             type=click.Choice(MySqlConfig.MySQLAuthType.values()),
-            default=MySqlConfig.MySQLAuthType.BASIC.value,
+            default=MySqlConfig.MySQLAuthType.BASIC,
         )
     )
     if mysql_db_config.auth_type == MySqlConfig.MySQLAuthType.BASIC:
         basic_auth_config = MySqlConfig.BasicConfig()
-        basic_auth_config.password = click.prompt(f"Password", hide_input=True)
+        basic_auth_config.password = click.prompt("Password", hide_input=True)
         mysql_db_config.basic_config = basic_auth_config
 
     elif mysql_db_config.auth_type == MySqlConfig.MySQLAuthType.VAULT:
@@ -176,12 +171,12 @@ def detect_or_create_stack(apigw_config: APIGatewayConfig):
     ).data
 
     if len(stacks) >= 1:
-        print(f"Auto-detected feature store stack(s) in tenancy:")
+        print("Auto-detected feature store stack(s) in tenancy:")
         for stack in stacks:
             _print_stack_detail(stack)
     choices = {"1": "new", "2": "existing"}
     stack_provision_method = click.prompt(
-        f"Select stack provisioning method:\n1.Create new stack\n2.Existing stack\n",
+        "Select stack provisioning method:\n1.Create new stack\n2.Existing stack\n",
         type=click.Choice(list(choices.keys())),
         show_choices=False,
     )
@@ -240,20 +235,20 @@ def get_api_gw_details(compartment_id: str) -> APIGatewayConfig:
 
 
 def get_nlb_id_from_service(service: "V1Service", apigw_config: APIGatewayConfig):
-    status: "V1ServiceStatus" = service.status
-    lb_status: "V1LoadBalancerStatus" = status.load_balancer
-    lb_ingress: "V1LoadBalancerIngress" = lb_status.ingress[0]
+    status: V1ServiceStatus = service.status
+    lb_status: V1LoadBalancerStatus = status.load_balancer
+    lb_ingress: V1LoadBalancerIngress = lb_status.ingress[0]
     resource_client = OCIClientFactory(**authutil.default_signer()).create_client(
         oci.resource_search.ResourceSearchClient
     )
     search_details = oci.resource_search.models.FreeTextSearchDetails()
     search_details.matching_context_type = "NONE"
     search_details.text = lb_ingress.ip
-    resources: List[
-        oci.resource_search.models.ResourceSummary
-    ] = resource_client.search_resources(
-        search_details, tenant_id=apigw_config.root_compartment_id
-    ).data.items
+    resources: List[oci.resource_search.models.ResourceSummary] = (
+        resource_client.search_resources(
+            search_details, tenant_id=apigw_config.root_compartment_id
+        ).data.items
+    )
     private_ips = list(filter(lambda obj: obj.resource_type == "PrivateIp", resources))
     if len(private_ips) != 1:
         return click.prompt(
@@ -264,12 +259,12 @@ def get_nlb_id_from_service(service: "V1Service", apigw_config: APIGatewayConfig
         nlb_client = OCIClientFactory(**authutil.default_signer()).create_client(
             oci.network_load_balancer.NetworkLoadBalancerClient
         )
-        nlbs: List[
-            oci.network_load_balancer.models.NetworkLoadBalancerSummary
-        ] = nlb_client.list_network_load_balancers(
-            compartment_id=nlb_private_ip.compartment_id,
-            display_name=nlb_private_ip.display_name,
-        ).data.items
+        nlbs: List[oci.network_load_balancer.models.NetworkLoadBalancerSummary] = (
+            nlb_client.list_network_load_balancers(
+                compartment_id=nlb_private_ip.compartment_id,
+                display_name=nlb_private_ip.display_name,
+            ).data.items
+        )
         if len(nlbs) != 1:
             return click.prompt(
                 f"Please enter OCID of load balancer associated with ip: {lb_ingress.ip}"

ads/opctl/operator/lowcode/forecast/__main__.py

@@ -17,6 +17,7 @@ from ads.opctl.operator.common.utils import _parse_input_args
 
 from .operator_config import ForecastOperatorConfig
 from .model.forecast_datasets import ForecastDatasets
+from .whatifserve import ModelDeploymentManager
 
 
 def operate(operator_config: ForecastOperatorConfig) -> None:
@@ -27,6 +28,15 @@ def operate(operator_config: ForecastOperatorConfig) -> None:
     ForecastOperatorModelFactory.get_model(
         operator_config, datasets
     ).generate_report()
+    # saving to model catalog
+    spec = operator_config.spec
+    if spec.what_if_analysis and datasets.additional_data:
+        mdm = ModelDeploymentManager(spec, datasets.additional_data)
+        mdm.save_to_catalog()
+        if spec.what_if_analysis.model_deployment:
+            mdm.create_deployment()
+        mdm.save_deployment_info()
+
 
 def verify(spec: Dict, **kwargs: Dict) -> bool:
     """Verifies the forecasting operator config."""