oracle-ads 2.12.11__py3-none-any.whl → 2.13.1rc0__py3-none-any.whl

ads/opctl/cmds.py

@@ -1,7 +1,6 @@
 #!/usr/bin/env python
-# -*- coding: utf-8; -*-
 
-# Copyright (c) 2022, 2024 Oracle and/or its affiliates.
+# Copyright (c) 2022, 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 configparser
@@ -11,13 +10,10 @@ from typing import Dict, List, Union
 import click
 import fsspec
 import yaml
-from ads.opctl.backend.marketplace.local_marketplace import (
-    LocalMarketplaceOperatorBackend,
-)
 
 import ads
 from ads.common.auth import AuthContext, AuthType
-from ads.common.extended_enum import ExtendedEnumMeta
+from ads.common.extended_enum import ExtendedEnum
 from ads.common.oci_datascience import DSCNotebookSession
 from ads.opctl.backend.ads_dataflow import DataFlowBackend
 from ads.opctl.backend.ads_ml_job import MLJobBackend, MLJobDistributedBackend
@@ -30,6 +26,9 @@ from ads.opctl.backend.local import (
     LocalOperatorBackend,
     LocalPipelineBackend,
 )
+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.config.resolver import ConfigResolver
@@ -64,7 +63,7 @@ from ads.opctl.operator.common.backend_factory import (
 from ads.opctl.utils import get_service_pack_prefix, is_in_notebook_session
 
 
-class DataScienceResource(str, metaclass=ExtendedEnumMeta):
+class DataScienceResource(ExtendedEnum):
     JOB = "datasciencejob"
     DATAFLOW = "dataflowapplication"
     PIPELINE = "datasciencepipeline"
@@ -72,7 +71,7 @@ class DataScienceResource(str, metaclass=ExtendedEnumMeta):
     MODEL = "datasciencemodel"
 
 
-class DataScienceResourceRun(str, metaclass=ExtendedEnumMeta):
+class DataScienceResourceRun(ExtendedEnum):
     JOB_RUN = "datasciencejobrun"
     DATAFLOW_RUN = "dataflowrun"
     PIPELINE_RUN = "datasciencepipelinerun"
@@ -100,18 +99,18 @@ DATA_SCIENCE_RESOURCE_RUN_BACKEND_MAP = {
 
 class _BackendFactory:
     BACKENDS_MAP = {
-        BACKEND_NAME.JOB.value: MLJobBackend,
-        BACKEND_NAME.DATAFLOW.value: DataFlowBackend,
-        BACKEND_NAME.PIPELINE.value: PipelineBackend,
-        BACKEND_NAME.MODEL_DEPLOYMENT.value: ModelDeploymentBackend,
-        BACKEND_NAME.OPERATOR_LOCAL.value: LocalOperatorBackend,
-        BACKEND_NAME.MARKETPLACE.value: LocalMarketplaceOperatorBackend,
+        BACKEND_NAME.JOB: MLJobBackend,
+        BACKEND_NAME.DATAFLOW: DataFlowBackend,
+        BACKEND_NAME.PIPELINE: PipelineBackend,
+        BACKEND_NAME.MODEL_DEPLOYMENT: ModelDeploymentBackend,
+        BACKEND_NAME.OPERATOR_LOCAL: LocalOperatorBackend,
+        BACKEND_NAME.MARKETPLACE: LocalMarketplaceOperatorBackend,
     }
 
     LOCAL_BACKENDS_MAP = {
-        BACKEND_NAME.JOB.value: LocalBackend,
-        BACKEND_NAME.PIPELINE.value: LocalPipelineBackend,
-        BACKEND_NAME.MODEL_DEPLOYMENT.value: LocalModelDeploymentBackend,
+        BACKEND_NAME.JOB: LocalBackend,
+        BACKEND_NAME.PIPELINE: LocalPipelineBackend,
+        BACKEND_NAME.MODEL_DEPLOYMENT: LocalModelDeploymentBackend,
     }
 
     def __init__(self, config: Dict):
@@ -120,14 +119,14 @@ class _BackendFactory:
         if self._backend is None:
             raise RuntimeError("Please specify backend.")
         elif (
-            self._backend != BACKEND_NAME.LOCAL.value
+            self._backend != BACKEND_NAME.LOCAL
             and self._backend not in self.BACKENDS_MAP
         ):
             raise NotImplementedError(f"backend {self._backend} is not implemented.")
 
     @property
     def backend(self):
-        if self._backend == BACKEND_NAME.LOCAL.value:
+        if self._backend == BACKEND_NAME.LOCAL:
             kind = self.config.get("kind") or self.config["execution"].get("kind")
             if kind not in self.LOCAL_BACKENDS_MAP:
                 options = [backend for backend in self.LOCAL_BACKENDS_MAP.keys()]
@@ -194,10 +193,7 @@ def run(config: Dict, **kwargs) -> Dict:
         except RuntimeError:
             pass
 
-        if (
-            p.config["kind"] != BACKEND_NAME.LOCAL.value
-            and p.config["kind"] != "distributed"
-        ):
+        if p.config["kind"] != BACKEND_NAME.LOCAL and p.config["kind"] != "distributed":
             p.config["execution"]["backend"] = p.config["kind"]
             return _BackendFactory(p.config).backend.apply()
     else:
@@ -226,9 +222,9 @@ def run(config: Dict, **kwargs) -> Dict:
             docker_build_cmd(ini)
         config = update_image(config, ini)
 
-        if mode == BACKEND_NAME.LOCAL.value:
+        if mode == BACKEND_NAME.LOCAL:
             print(
-                "\u26A0 Docker Image: "
+                "\u26a0 Docker Image: "
                 + ini.get("main", "registry")
                 + ":"
                 + ini.get("main", "tag")
@@ -238,7 +234,7 @@ def run(config: Dict, **kwargs) -> Dict:
 
             backend = LocalBackendDistributed(config)
             backend.run()
-        elif mode == BACKEND_NAME.DATAFLOW.value:
+        elif mode == BACKEND_NAME.DATAFLOW:
             raise RuntimeError(
                 "backend operator for distributed training can either be local or job"
             )
@@ -283,7 +279,7 @@ def run(config: Dict, **kwargs) -> Dict:
             }
             for r, b in resource_to_backend.items():
                 if r in p.config["execution"]["ocid"]:
-                    p.config["execution"]["backend"] = b.value
+                    p.config["execution"]["backend"] = b
         else:
             p.step(ConfigResolver).step(ConfigValidator)
             # spec may have changed during validation step (e.g. defaults filled in)
@@ -592,7 +588,7 @@ def configure() -> None:
 
     print("==== Setting configuration for Data Science Jobs ====")
     if click.confirm(
-        f"Do you want to set up or update Data Science Jobs configuration?",
+        "Do you want to set up or update Data Science Jobs configuration?",
         default=True,
     ):
         required_fields = [
@@ -622,7 +618,7 @@ def configure() -> None:
 
     print("==== Setting configuration for OCI Data Flow ====")
     if click.confirm(
-        f"Do you want to set up or update OCI Data Flow configuration?", default=True
+        "Do you want to set up or update OCI Data Flow configuration?", default=True
     ):
         required_fields = [
             ("compartment_id", ""),
@@ -652,7 +648,7 @@ def configure() -> None:
 
     print("==== Setting configuration for OCI ML Pipeline ====")
     if click.confirm(
-        f"Do you want to set up or update OCI ML Pipeline configuration?", default=True
+        "Do you want to set up or update OCI ML Pipeline configuration?", default=True
     ):
         required_fields = [
             ("compartment_id", ""),
@@ -674,7 +670,7 @@ def configure() -> None:
 
     print("==== Setting configuration for Data Science Model Deployment ====")
     if click.confirm(
-        f"Do you want to set up or update Data Science Model Deployment configuration?",
+        "Do you want to set up or update Data Science Model Deployment configuration?",
         default=True,
     ):
         required_fields = [
@@ -704,7 +700,7 @@ def configure() -> None:
 
     print("==== Setting configuration for local backend ====")
     if click.confirm(
-        f"Do you want to set up or update local backend configuration?", default=True
+        "Do you want to set up or update local backend configuration?", default=True
     ):
         required_fields = [
             ("max_parallel_containers", str(min(os.cpu_count(), 4))),

ads/opctl/config/merger.py

@@ -1,6 +1,6 @@
 #!/usr/bin/env python
 
-# Copyright (c) 2022, 2024 Oracle and/or its affiliates.
+# Copyright (c) 2022, 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 os
@@ -190,11 +190,11 @@ class ConfigMerger(ConfigProcessor):
     def _get_service_config(self, oci_profile: str, ads_config_folder: str) -> Dict:
         backend = self.config["execution"].get("backend", None)
         backend_config = {
-            BACKEND_NAME.JOB.value: ADS_JOBS_CONFIG_FILE_NAME,
-            BACKEND_NAME.DATAFLOW.value: ADS_DATAFLOW_CONFIG_FILE_NAME,
-            BACKEND_NAME.PIPELINE.value: ADS_ML_PIPELINE_CONFIG_FILE_NAME,
-            BACKEND_NAME.LOCAL.value: ADS_LOCAL_BACKEND_CONFIG_FILE_NAME,
-            BACKEND_NAME.MODEL_DEPLOYMENT.value: ADS_MODEL_DEPLOYMENT_CONFIG_FILE_NAME,
+            BACKEND_NAME.JOB: ADS_JOBS_CONFIG_FILE_NAME,
+            BACKEND_NAME.DATAFLOW: ADS_DATAFLOW_CONFIG_FILE_NAME,
+            BACKEND_NAME.PIPELINE: ADS_ML_PIPELINE_CONFIG_FILE_NAME,
+            BACKEND_NAME.LOCAL: ADS_LOCAL_BACKEND_CONFIG_FILE_NAME,
+            BACKEND_NAME.MODEL_DEPLOYMENT: ADS_MODEL_DEPLOYMENT_CONFIG_FILE_NAME,
         }
         config_file = backend_config.get(backend, ADS_JOBS_CONFIG_FILE_NAME)
 
@@ -213,10 +213,7 @@ class ConfigMerger(ConfigProcessor):
     def _config_flex_shape_details(self):
         infrastructure = self.config["infrastructure"]
         backend = self.config["execution"].get("backend", None)
-        if (
-            backend == BACKEND_NAME.JOB.value
-            or backend == BACKEND_NAME.MODEL_DEPLOYMENT.value
-        ):
+        if backend == BACKEND_NAME.JOB or backend == BACKEND_NAME.MODEL_DEPLOYMENT:
             shape_name = infrastructure.get("shape_name", "")
             if shape_name.endswith(".Flex"):
                 if (
@@ -231,7 +228,7 @@ class ConfigMerger(ConfigProcessor):
                     "ocpus": infrastructure.pop("ocpus"),
                     "memory_in_gbs": infrastructure.pop("memory_in_gbs"),
                 }
-        elif backend == BACKEND_NAME.DATAFLOW.value:
+        elif backend == BACKEND_NAME.DATAFLOW:
             executor_shape = infrastructure.get("executor_shape", "")
             driver_shape = infrastructure.get("driver_shape", "")
             data_flow_shape_config_details = [

ads/opctl/config/resolver.py

@@ -1,36 +1,34 @@
 #!/usr/bin/env python
-# -*- coding: utf-8; -*-
 
-# Copyright (c) 2022, 2023 Oracle and/or its affiliates.
+# Copyright (c) 2022, 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 base64
+import glob
 import json
 import os
-from typing import Dict
-from typing import Tuple
+from typing import Dict, Tuple
 
 import yaml
-import glob
 
 from ads.common.auth import create_signer
+from ads.common.decorator.runtime_dependency import (
+    OptionalDependency,
+    runtime_dependency,
+)
 from ads.opctl import logger
 from ads.opctl.config.base import ConfigProcessor
 from ads.opctl.config.utils import NotSupportedError, convert_notebook
 from ads.opctl.constants import (
+    BACKEND_NAME,
     ML_JOB_GPU_IMAGE,
     ML_JOB_IMAGE,
-    BACKEND_NAME,
 )
 from ads.opctl.utils import (
+    get_namespace,
+    get_region_key,
     list_ads_operators,
     parse_conda_uri,
-    get_region_key,
-    get_namespace,
-)
-from ads.common.decorator.runtime_dependency import (
-    runtime_dependency,
-    OptionalDependency,
 )
 
 
@@ -91,7 +89,7 @@ class ConfigResolver(ConfigProcessor):
         if not (
             self.config["execution"].get("conda_slug")
             or self.config["execution"].get("image")
-            or self.config["execution"]["backend"] == BACKEND_NAME.DATAFLOW.value
+            or self.config["execution"]["backend"] == BACKEND_NAME.DATAFLOW
         ):
             raise ValueError(
                 "If not running an operator, conda pack info or image name needs to be given."
@@ -134,21 +132,18 @@ class ConfigResolver(ConfigProcessor):
                 raise FileNotFoundError(
                     f"{self.config['execution']['source_folder']} is not found."
                 )
-        else:
-            if self._is_ads_operator():
-                curr_dir = os.path.dirname(os.path.abspath(__file__))
-                self.config["execution"]["source_folder"] = os.path.normpath(
-                    os.path.join(
-                        curr_dir,
-                        "..",
-                        "operators",
-                        inflection.underscore(
-                            self.config["execution"]["operator_name"]
-                        ),
-                    )
+        elif self._is_ads_operator():
+            curr_dir = os.path.dirname(os.path.abspath(__file__))
+            self.config["execution"]["source_folder"] = os.path.normpath(
+                os.path.join(
+                    curr_dir,
+                    "..",
+                    "operators",
+                    inflection.underscore(self.config["execution"]["operator_name"]),
                 )
-            else:
-                self.config["execution"]["source_folder"] = None
+            )
+        else:
+            self.config["execution"]["source_folder"] = None
 
     def _resolve_entry_script(self) -> None:
         # this should be run after _resolve_source_folder_path
@@ -263,9 +258,9 @@ class ConfigResolver(ConfigProcessor):
                     self.config["infrastructure"]["docker_registry"], image
                 )
             else:
-                self.config["execution"][
-                    "image"
-                ] = f"{region_key}.ocir.io/{namespace}/{image}"
+                self.config["execution"]["image"] = (
+                    f"{region_key}.ocir.io/{namespace}/{image}"
+                )
 
     def _resolve_env_vars(self) -> None:
         env_vars = self.config["execution"].get("env_vars", {})

ads/opctl/forecast.py

@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+
+# Copyright (c) 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.opctl.operator.lowcode.forecast.__main__ import operate
+from ads.opctl.operator.lowcode.forecast.operator_config import ForecastOperatorConfig
+
+if __name__ == "__main__":
+    config = ForecastOperatorConfig()
+    operate(config)

ads/opctl/operator/cli.py

@@ -1,25 +1,24 @@
 #!/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/
 
+import logging
 from typing import Any, Dict
 
 import click
 import fsspec
 import yaml
-import logging
-from ads.opctl.operator.common.utils import default_signer
+
 from ads.common.auth import AuthType
 from ads.common.object_storage_details import ObjectStorageDetails
+from ads.opctl import logger
 from ads.opctl.constants import BACKEND_NAME, RUNTIME_TYPE
 from ads.opctl.decorator.common import click_options, with_auth, with_click_unknown_args
+from ads.opctl.operator.common.utils import default_signer
 from ads.opctl.utils import suppress_traceback
-from ads.opctl import logger
 
 from .__init__ import __operators__
-from .cmd import run as cmd_run
 from .cmd import build_conda as cmd_build_conda
 from .cmd import build_image as cmd_build_image
 from .cmd import create as cmd_create
@@ -28,6 +27,7 @@ from .cmd import init as cmd_init
 from .cmd import list as cmd_list
 from .cmd import publish_conda as cmd_publish_conda
 from .cmd import publish_image as cmd_publish_image
+from .cmd import run as cmd_run
 from .cmd import verify as cmd_verify
 
 DEBUG_OPTION = (
@@ -113,7 +113,7 @@ def info(debug: bool, **kwargs: Dict[str, Any]) -> None:
 )
 @click.option(
     "--output",
-    help=f"The folder name to save the resulting specification templates.",
+    help="The folder name to save the resulting specification templates.",
     required=False,
     default=None,
 )
@@ -292,9 +292,9 @@ def publish_conda(debug: bool, **kwargs: Dict[str, Any]) -> None:
     "-b",
     help=(
         "Backend name or the path to the operator's backend config YAML file. "
-        f"\n\nExample 1:\n\n`ads operator run -f operator.yaml -b {BACKEND_NAME.LOCAL.value}`\n\n"
+        f"\n\nExample 1:\n\n`ads operator run -f operator.yaml -b {BACKEND_NAME.LOCAL}`\n\n"
         "Supported backend names: "
-        f"{(BACKEND_NAME.JOB.value,BACKEND_NAME.JOB.value + '.' + RUNTIME_TYPE.CONTAINER.value,BACKEND_NAME.DATAFLOW.value,BACKEND_NAME.LOCAL.value,BACKEND_NAME.LOCAL.value + '.'+ RUNTIME_TYPE.CONTAINER.value,)}. "
+        f"{(BACKEND_NAME.JOB,BACKEND_NAME.JOB + '.' + RUNTIME_TYPE.CONTAINER,BACKEND_NAME.DATAFLOW,BACKEND_NAME.LOCAL,BACKEND_NAME.LOCAL + '.'+ RUNTIME_TYPE.CONTAINER,)}. "
         "However some operators may support only a subset of these backends."
         "\n\nExample 2:\n\n`ads operator run -f operator.yaml -b backend.yaml`\n\n"
         "Use the `ads operator init --help` command to generate the operator's specification "

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"