sagemaker-core 0.1.3__py3-none-any.whl → 1.0.1__py3-none-any.whl

sagemaker_core/main/user_agent.py

@@ -0,0 +1,77 @@
+# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License"). You
+# may not use this file except in compliance with the License. A copy of
+# the License is located at
+#
+#     http://aws.amazon.com/apache2.0/
+#
+# or in the "license" file accompanying this file. This file is
+# distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF
+# ANY KIND, either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+from __future__ import absolute_import
+
+import json
+import os
+
+import importlib_metadata
+
+SagemakerCore_PREFIX = "AWS-SageMakerCore"
+STUDIO_PREFIX = "AWS-SageMaker-Studio"
+NOTEBOOK_PREFIX = "AWS-SageMaker-Notebook-Instance"
+
+NOTEBOOK_METADATA_FILE = "/etc/opt/ml/sagemaker-notebook-instance-version.txt"
+STUDIO_METADATA_FILE = "/opt/ml/metadata/resource-metadata.json"
+
+SagemakerCore_VERSION = "v0.1.6"
+
+
+def process_notebook_metadata_file() -> str:
+    """Check if the platform is SageMaker Notebook, if yes, return the InstanceType
+
+    Returns:
+        str: The InstanceType of the SageMaker Notebook if it exists, otherwise None
+    """
+    if os.path.exists(NOTEBOOK_METADATA_FILE):
+        with open(NOTEBOOK_METADATA_FILE, "r") as sagemaker_nbi_file:
+            return sagemaker_nbi_file.read().strip()
+
+    return None
+
+
+def process_studio_metadata_file() -> str:
+    """Check if the platform is SageMaker Studio, if yes, return the AppType
+
+    Returns:
+        str: The AppType of the SageMaker Studio if it exists, otherwise None
+    """
+    if os.path.exists(STUDIO_METADATA_FILE):
+        with open(STUDIO_METADATA_FILE, "r") as sagemaker_studio_file:
+            metadata = json.load(sagemaker_studio_file)
+            return metadata.get("AppType")
+
+    return None
+
+
+def get_user_agent_extra_suffix() -> str:
+    """Get the user agent extra suffix string specific to SageMakerCore
+
+    Adhers to new boto recommended User-Agent 2.0 header format
+
+    Returns:
+        str: The user agent extra suffix string to be appended
+    """
+    suffix = "lib/{}#{}".format(SagemakerCore_PREFIX, SagemakerCore_VERSION)
+
+    # Get the notebook instance type and prepend it to the user agent string if exists
+    notebook_instance_type = process_notebook_metadata_file()
+    if notebook_instance_type:
+        suffix = "{} md/{}#{}".format(suffix, NOTEBOOK_PREFIX, notebook_instance_type)
+
+    # Get the studio app type and prepend it to the user agent string if exists
+    studio_app_type = process_studio_metadata_file()
+    if studio_app_type:
+        suffix = "{} md/{}#{}".format(suffix, STUDIO_PREFIX, studio_app_type)
+
+    return suffix

sagemaker_core/{generated → main}/utils.py

@@ -15,14 +15,174 @@ import datetime
 import logging
 import os
 import re
+import subprocess
 
 from boto3.session import Session
 from botocore.config import Config
-from typing import TypeVar, Generic, Type
-from sagemaker_core.code_injection.codec import transform
+from rich import reconfigure
+from rich.console import Console
+from rich.logging import RichHandler
+from rich.style import Style
+from rich.theme import Theme
+from rich.traceback import install
+from typing import Any, Dict, List, TypeVar, Generic, Type
+from sagemaker_core.main.code_injection.codec import transform
+from sagemaker_core.main.code_injection.constants import Color
+from sagemaker_core.main.user_agent import get_user_agent_extra_suffix
+
+
+def add_indent(text, num_spaces=4):
+    """
+    Add customizable indent spaces to a given text.
+    Parameters:
+        text (str): The text to which the indent spaces will be added.
+        num_spaces (int): Number of spaces to be added for each level of indentation. Default is 4.
+    Returns:
+        str: The text with added indent spaces.
+    """
+    indent = " " * num_spaces
+    lines = text.split("\n")
+    indented_text = "\n".join(indent + line for line in lines)
+    return indented_text.rstrip(" ")
+
+
+def clean_documentaion(documentation):
+    documentation = re.sub(r"<\/?p>", "", documentation)
+    documentation = re.sub(r"<\/?code>", "'", documentation)
+    return documentation
+
+
+def convert_to_snake_case(entity_name):
+    """
+    Convert a string to snake_case.
+    Args:
+        entity_name (str): The string to convert.
+    Returns:
+        str: The converted string in snake_case.
+    """
+    snake_case = re.sub("(.)([A-Z][a-z]+)", r"\1_\2", entity_name)
+    return re.sub("([a-z0-9])([A-Z])", r"\1_\2", snake_case).lower()
+
+
+def snake_to_pascal(snake_str):
+    """
+    Convert a snake_case string to PascalCase.
+    Args:
+        snake_str (str): The snake_case string to be converted.
+    Returns:
+        str: The PascalCase string.
+    """
+    components = snake_str.split("_")
+    return "".join(x.title() for x in components[0:])
+
+
+def reformat_file_with_black(filename):
+    try:
+        # Run black with specific options using subprocess
+        subprocess.run(["black", "-l", "100", filename], check=True)
+        print(f"File '{filename}' reformatted successfully.")
+    except subprocess.CalledProcessError as e:
+        print(f"An error occurred while reformatting '{filename}': {e}")
+
+
+def remove_html_tags(text):
+    clean = re.compile("<.*?>")
+    return re.sub(clean, "", text)
+
+
+def escape_special_rst_characters(text):
+    # List of special characters that need to be escaped in reStructuredText
+    special_characters = ["*", "|"]
+
+    for char in special_characters:
+        # Use a regex to find the special character if preceded by a space
+        pattern = rf"(?<=\s){re.escape(char)}"
+        text = re.sub(pattern, rf"\\{char}", text)
+
+    return text
+
+
+def get_textual_rich_theme() -> Theme:
+    """
+    Get a textual rich theme with customized styling.
+
+    Returns:
+        Theme: A textual rich theme
+    """
+    return Theme(
+        {
+            "logging.level.info": Style(color=Color.BLUE.value, bold=True),
+            "logging.level.debug": Style(color=Color.GREEN.value, bold=True),
+            "logging.level.warning": Style(color=Color.YELLOW.value, bold=True),
+            "logging.level.error": Style(color=Color.RED.value, bold=True),
+            "logging.keyword": Style(color=Color.YELLOW.value, bold=True),
+            "repr.attrib_name": Style(color=Color.YELLOW.value, italic=False),
+            "repr.attrib_value": Style(color=Color.PURPLE.value, italic=False),
+            "repr.bool_true": Style(color=Color.GREEN.value, italic=True),
+            "repr.bool_false": Style(color=Color.RED.value, italic=True),
+            "repr.call": Style(color=Color.PURPLE.value, bold=True),
+            "repr.none": Style(color=Color.PURPLE.value, italic=True),
+            "repr.str": Style(color=Color.GREEN.value),
+            "repr.path": Style(color=Color.PURPLE.value),
+            "repr.filename": Style(color=Color.PURPLE.value),
+            "repr.url": Style(color=Color.BLUE.value, underline=True),
+            "repr.tag_name": Style(color=Color.PURPLE.value, bold=True),
+            "repr.ipv4": Style.null(),
+            "repr.ipv6": Style.null(),
+            "repr.eui48": Style.null(),
+            "repr.eui64": Style.null(),
+            "json.bool_true": Style(color=Color.GREEN.value, italic=True),
+            "json.bool_false": Style(color=Color.RED.value, italic=True),
+            "json.null": Style(color=Color.PURPLE.value, italic=True),
+            "json.str": Style(color=Color.GREEN.value),
+            "json.key": Style(color=Color.BLUE.value, bold=True),
+            "traceback.error": Style(color=Color.BRIGHT_RED.value, italic=True),
+            "traceback.border": Style(color=Color.BRIGHT_RED.value),
+            "traceback.title": Style(color=Color.BRIGHT_RED.value, bold=True),
+        }
+    )
+
+
+textual_rich_console_and_traceback_enabled = False
+
+
+def enable_textual_rich_console_and_traceback():
+    """
+    Reconfigure the global textual rich console with the customized theme
+        and enable textual rich error traceback
+    """
+    global textual_rich_console_and_traceback_enabled
+    if not textual_rich_console_and_traceback_enabled:
+        theme = get_textual_rich_theme()
+        reconfigure(theme=theme)
+        console = Console(theme=theme)
+        install(console=console)
+        textual_rich_console_and_traceback_enabled = True
+
+
+def get_textual_rich_logger(name: str, log_level: str = "INFO") -> logging.Logger:
+    """
+    Get a logger with textual rich handler.
+
+    Args:
+        name (str): The name of the logger
+        log_level (str): The log level to set.
+            Accepted values are: "DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL".
+            Defaults to the value of "INFO".
+
+    Return:
+        logging.Logger: A textial rich logger.
+
+    """
+    enable_textual_rich_console_and_traceback()
+    handler = RichHandler(markup=True)
+    logging.basicConfig(level=getattr(logging, log_level), handlers=[handler])
+    logger = logging.getLogger(name)
 
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+    return logger
+
+
+logger = get_textual_rich_logger(__name__)
 
 T = TypeVar("T")
 
@@ -57,10 +217,7 @@ def configure_logging(log_level=None):
     # reset any currently associated handlers with log level
     for handler in _logger.handlers:
         _logger.removeHandler(handler)
-    console_handler = logging.StreamHandler()
-    console_handler.setFormatter(
-        logging.Formatter("%(asctime)s : %(levelname)s : %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
-    )
+    console_handler = RichHandler(markup=True)
     _logger.addHandler(console_handler)
 
 
@@ -182,10 +339,11 @@ class SageMakerClient(metaclass=SingletonMeta):
             logger.warning("No config provided. Using default config.")
             config = Config(retries={"max_attempts": 10, "mode": "standard"})
 
+        self.config = Config(user_agent_extra=get_user_agent_extra_suffix())
         self.session = session
         self.region_name = region_name
         self.service_name = service_name
-        self.client = session.client(service_name, region_name, config=config)
+        self.client = session.client(service_name, region_name, config=self.config)
 
 
 class SageMakerRuntimeClient(metaclass=SingletonMeta):
@@ -216,10 +374,11 @@ class SageMakerRuntimeClient(metaclass=SingletonMeta):
             logger.warning("No config provided. Using default config.")
             config = Config(retries={"max_attempts": 10, "mode": "standard"})
 
+        self.config = Config(user_agent_extra=get_user_agent_extra_suffix())
         self.session = session
         self.region_name = region_name
         self.service_name = service_name
-        self.client = session.client(service_name, region_name, config=config)
+        self.client = session.client(service_name, region_name, config=self.config)
 
 
 class ResourceIterator(Generic[T]):
@@ -312,3 +471,80 @@ class ResourceIterator(Generic[T]):
                 raise StopIteration
 
             return self.__next__()
+
+
+def serialize(value: Any) -> Any:
+    """
+    Serialize an object recursively by converting all objects to JSON-serializable types
+
+    Args:
+       value (Any): The object to be serialized
+
+    Returns:
+        Any: The serialized object
+    """
+    if isinstance(value, Unassigned):
+        return None
+    elif isinstance(value, Dict):
+        # if the value is a dict, use _serialize_dict() to serialize it recursively
+        return _serialize_dict(value)
+    elif isinstance(value, List):
+        # if the value is a dict, use _serialize_list() to serialize it recursively
+        return _serialize_list(value)
+    elif is_not_primitive(value):
+        # if the value is a dict, use _serialize_shape() to serialize it recursively
+        return _serialize_shape(value)
+    else:
+        return value
+
+
+def _serialize_dict(value: Dict) -> dict:
+    """
+    Serialize all values in a dict recursively
+
+    Args:
+       value (dict): The dict to be serialized
+
+    Returns:
+        dict: The serialized dict
+    """
+    serialized_dict = {}
+    for k, v in value.items():
+        if serialize_result := serialize(v):
+            serialized_dict.update({k: serialize_result})
+    return serialized_dict
+
+
+def _serialize_list(value: List) -> list:
+    """
+    Serialize all objects in a list
+
+    Args:
+       value (list): The dict to be serialized
+
+    Returns:
+        list: The serialized list
+    """
+    serialized_list = []
+    for v in value:
+        if serialize_result := serialize(v):
+            serialized_list.append(serialize_result)
+    return serialized_list
+
+
+def _serialize_shape(value: Any) -> dict:
+    """
+    Serialize a shape object defined in resource.py or shape.py to a dict
+
+    Args:
+       value (Any): The shape to be serialized
+
+    Returns:
+        dict: The dict of serialized shape
+    """
+    serialized_dict = {}
+    for k, v in vars(value).items():
+        if serialize_result := serialize(v):
+            key = snake_to_pascal(k) if is_snake_case(k) else k
+            serialized_dict.update({key[0].upper() + key[1:]: serialize_result})
+    return serialized_dict

sagemaker_core/resources/__init__.py

@@ -0,0 +1 @@
+from ..main.resources import *

sagemaker_core/shapes/__init__.py

@@ -0,0 +1 @@
+from ..main.shapes import *

sagemaker_core/tools/__init__.py

@@ -1 +1 @@
-from ..code_injection.codec import pascal_to_snake
+from ..main.code_injection.codec import pascal_to_snake

sagemaker_core/tools/codegen.py

@@ -11,12 +11,12 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Generates the code for the service model."""
+from sagemaker_core.main.utils import reformat_file_with_black
 from sagemaker_core.tools.shapes_codegen import ShapesCodeGen
 from sagemaker_core.tools.resources_codegen import ResourcesCodeGen
 from typing import Optional
 
 from sagemaker_core.tools.data_extractor import ServiceJsonData, load_service_jsons
-from sagemaker_core.util.util import reformat_file_with_black
 
 
 def generate_code(
@@ -45,7 +45,6 @@ def generate_code(
     )
 
     shapes_code_gen.generate_shapes()
-    resources_code_gen.generate_resources()
     reformat_file_with_black(".")
 
 

sagemaker_core/tools/constants.py

@@ -44,7 +44,7 @@ BASIC_JSON_TYPES_TO_PYTHON_TYPES = {
 
 BASIC_RETURN_TYPES = {"str", "int", "bool", "float", "datetime.datetime"}
 
-SHAPE_DAG_FILE_PATH = os.getcwd() + "/src/sagemaker_core/code_injection/shape_dag.py"
+SHAPE_DAG_FILE_PATH = os.getcwd() + "/src/sagemaker_core/main/code_injection/shape_dag.py"
 PYTHON_TYPES_TO_BASIC_JSON_TYPES = {
     "str": "string",
     "int": "integer",
@@ -68,13 +68,8 @@ LICENCES_STRING = """
 # language governing permissions and limitations under the License.
 """
 
-BASIC_IMPORTS_STRING = """
-import logging
-"""
-
 LOGGER_STRING = """
-logging.basicConfig(level=logging.INFO)
-logger = logging.getLogger(__name__)
+logger = get_textual_rich_logger(__name__)
 
 """
 
@@ -85,7 +80,7 @@ ADDITIONAL_OPERATION_FILE_PATH = (
 SERVICE_JSON_FILE_PATH = os.getcwd() + "/sample/sagemaker/2017-07-24/service-2.json"
 RUNTIME_SERVICE_JSON_FILE_PATH = os.getcwd() + "/sample/sagemaker-runtime/2017-05-13/service-2.json"
 
-GENERATED_CLASSES_LOCATION = os.getcwd() + "/src/sagemaker_core/generated"
+GENERATED_CLASSES_LOCATION = os.getcwd() + "/src/sagemaker_core/main"
 UTILS_CODEGEN_FILE_NAME = "utils.py"
 INTELLIGENT_DEFAULTS_HELPER_CODEGEN_FILE_NAME = "intelligent_defaults_helper.py"
 

sagemaker_core/tools/method.py

@@ -1,6 +1,6 @@
 from enum import Enum
 
-from sagemaker_core.util.util import remove_html_tags
+from sagemaker_core.main.utils import remove_html_tags
 
 
 class MethodType(Enum):

sagemaker_core/tools/resources_codegen.py

@@ -11,33 +11,32 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Generates the resource classes for the service model."""
-from collections import OrderedDict
-import logging
 from functools import lru_cache
 
 import os
 import json
-from sagemaker_core.code_injection.codec import pascal_to_snake
-from sagemaker_core.generated.config_schema import SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA
-from sagemaker_core.generated.exceptions import IntelligentDefaultsError
+from sagemaker_core.main.code_injection.codec import pascal_to_snake
+from sagemaker_core.main.config_schema import SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA
+from sagemaker_core.main.exceptions import IntelligentDefaultsError
+from sagemaker_core.main.utils import get_textual_rich_logger
 from sagemaker_core.tools.constants import (
     BASIC_RETURN_TYPES,
     GENERATED_CLASSES_LOCATION,
     RESOURCES_CODEGEN_FILE_NAME,
     LICENCES_STRING,
     TERMINAL_STATES,
-    BASIC_IMPORTS_STRING,
     LOGGER_STRING,
     CONFIG_SCHEMA_FILE_NAME,
     PYTHON_TYPES_TO_BASIC_JSON_TYPES,
     CONFIGURABLE_ATTRIBUTE_SUBSTRINGS,
 )
 from sagemaker_core.tools.method import Method, MethodType
-from sagemaker_core.util.util import (
+from sagemaker_core.main.utils import (
     add_indent,
     convert_to_snake_case,
     snake_to_pascal,
     remove_html_tags,
+    escape_special_rst_characters,
 )
 from sagemaker_core.tools.resources_extractor import ResourcesExtractor
 from sagemaker_core.tools.shapes_extractor import ShapesExtractor
@@ -82,8 +81,7 @@ from sagemaker_core.tools.data_extractor import (
     load_combined_operations_data,
 )
 
-logging.basicConfig(level=logging.INFO)
-log = logging.getLogger(__name__)
+log = get_textual_rich_logger(__name__)
 
 TYPE = "type"
 OBJECT = "object"
@@ -176,21 +174,26 @@ class ResourcesCodeGen:
         """
         # List of import statements
         imports = [
-            BASIC_IMPORTS_STRING,
             "import botocore",
             "import datetime",
             "import time",
             "import functools",
-            "from pprint import pprint",
             "from pydantic import validate_call",
-            "from typing import Dict, List, Literal, Optional, Union\n"
+            "from typing import Dict, List, Literal, Optional, Union, Any\n"
             "from boto3.session import Session",
-            "from sagemaker_core.code_injection.codec import transform",
-            "from sagemaker_core.generated.utils import SageMakerClient, SageMakerRuntimeClient, ResourceIterator,"
-            " Unassigned, snake_to_pascal, pascal_to_snake, is_not_primitive, is_not_str_dict, is_snake_case, is_primitive_list",
-            "from sagemaker_core.generated.intelligent_defaults_helper import load_default_configs_for_resource_name, get_config_value",
-            "from sagemaker_core.generated.shapes import *",
-            "from sagemaker_core.generated.exceptions import *",
+            "from rich.console import Group",
+            "from rich.live import Live",
+            "from rich.panel import Panel",
+            "from rich.progress import Progress, SpinnerColumn, TextColumn, TimeElapsedColumn",
+            "from rich.status import Status",
+            "from rich.style import Style",
+            "from sagemaker_core.main.code_injection.codec import transform",
+            "from sagemaker_core.main.code_injection.constants import Color",
+            "from sagemaker_core.main.utils import SageMakerClient, SageMakerRuntimeClient, ResourceIterator, Unassigned, get_textual_rich_logger, "
+            "snake_to_pascal, pascal_to_snake, is_not_primitive, is_not_str_dict, is_snake_case, is_primitive_list, serialize",
+            "from sagemaker_core.main.intelligent_defaults_helper import load_default_configs_for_resource_name, get_config_value",
+            "from sagemaker_core.main.shapes import *",
+            "from sagemaker_core.main.exceptions import *",
         ]
 
         formated_imports = "\n".join(imports)
@@ -344,6 +347,7 @@ class ResourcesCodeGen:
             class_attributes, class_attributes_string, attributes_and_documentation = (
                 class_attribute_info
             )
+
             # Start defining the class
             resource_class = f"class {resource_name}(Base):\n"
 
@@ -640,7 +644,8 @@ class ResourcesCodeGen:
                 else:
                     documentation_string += f"{attribute_snake}: {documentation}\n"
         documentation_string = add_indent(documentation_string)
-        return remove_html_tags(documentation_string)
+        documentation_string = remove_html_tags(documentation_string)
+        return escape_special_rst_characters(documentation_string)
 
     def _generate_create_method_args(
         self, operation_input_shape_name: str, resource_name: str
@@ -930,7 +935,7 @@ class ResourcesCodeGen:
                 decorator=decorator,
                 method_name="create",
                 method_args=add_indent("cls,\n", 4) + create_args,
-                return_type='Optional["resource_name"]',
+                return_type=f'Optional["{resource_name}"]',
                 serialize_operation_input=serialize_operation_input,
                 initialize_client=initialize_client,
                 call_operation_api=call_operation_api,
@@ -1758,7 +1763,7 @@ class ResourcesCodeGen:
         formatted_failed_block = FAILED_STATUS_ERROR_TEMPLATE.format(
             resource_name=resource_name, reason=failure_reason
         )
-        formatted_failed_block = add_indent(formatted_failed_block, 12)
+        formatted_failed_block = add_indent(formatted_failed_block, 16)
 
         formatted_method = WAIT_METHOD_TEMPLATE.format(
             terminal_resource_states=terminal_resource_states,
@@ -1792,7 +1797,7 @@ class ResourcesCodeGen:
             formatted_failed_block = FAILED_STATUS_ERROR_TEMPLATE.format(
                 resource_name=resource_name, reason=failure_reason
             )
-            formatted_failed_block = add_indent(formatted_failed_block, 8)
+            formatted_failed_block = add_indent(formatted_failed_block, 12)
 
         formatted_method = WAIT_FOR_STATUS_METHOD_TEMPLATE.format(
             resource_states=resource_states,
@@ -1826,10 +1831,10 @@ class ResourcesCodeGen:
             formatted_failed_block = DELETE_FAILED_STATUS_CHECK.format(
                 resource_name=resource_name, reason=failure_reason
             )
-            formatted_failed_block = add_indent(formatted_failed_block, 12)
+            formatted_failed_block = add_indent(formatted_failed_block, 16)
 
         if any(state.lower() == "deleted" for state in resource_states):
-            deleted_status_check = add_indent(DELETED_STATUS_CHECK, 12)
+            deleted_status_check = add_indent(DELETED_STATUS_CHECK, 16)
         else:
             deleted_status_check = ""
 
@@ -1962,9 +1967,6 @@ class ResourcesCodeGen:
         Input for generating the Schema is the service JSON that is already loaded in the class
 
         """
-        self.resources_extractor = ResourcesExtractor()
-        self.resources_plan = self.resources_extractor.get_resource_plan()
-
         resource_properties = {}
 
         for _, row in self.resources_plan.iterrows():
@@ -2073,7 +2075,7 @@ class ResourcesCodeGen:
                                 TYPE: self._get_json_schema_type_from_python_type(value) or value
                             }
             elif value.startswith("List") or value.startswith("Dict"):
-                log.info("Script does not currently support list of objects as configurable")
+                log.debug("Script does not currently support list of objects as configurable")
                 continue
             else:
                 class_attributes = self.shapes_extractor.generate_shape_members(value)

sagemaker_core/tools/resources_extractor.py

@@ -11,11 +11,11 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """A class for extracting resource information from a service JSON."""
-import logging
 from typing import Optional
 
 import pandas as pd
 
+from sagemaker_core.main.utils import get_textual_rich_logger
 from sagemaker_core.tools.constants import CLASS_METHODS, OBJECT_METHODS
 from sagemaker_core.tools.data_extractor import (
     load_additional_operations_data,
@@ -24,8 +24,7 @@ from sagemaker_core.tools.data_extractor import (
 )
 from sagemaker_core.tools.method import Method
 
-logging.basicConfig(level=logging.INFO)
-log = logging.getLogger(__name__)
+log = get_textual_rich_logger(__name__)
 """
 This class is used to extract the resources and its actions from the service-2.json file.
 """
@@ -175,9 +174,9 @@ class ResourcesExtractor:
 
         log.info(f"Total resource - {len(self.resources)}")
 
-        log.info(f"Total actions_under_resource - {len(self.actions_under_resource)}")
+        log.info(f"Supported actions - {len(self.actions_under_resource)}")
 
-        log.info(f"Unsupported actions: - {len(self.actions)}")
+        log.info(f"Unsupported actions - {len(self.actions)}")
 
         self._extract_resource_plan_as_dataframe()
 
@@ -368,6 +367,3 @@ class ResourcesExtractor:
             resource_methods (dict): The resource methods dict.
         """
         return self.resource_methods
-
-
-resource_extractor = ResourcesExtractor()