def generate_imports()

in src/sagemaker_core/tools/resources_codegen.py [0:0]

• 27 lines of code
• 2 McCabe index (conditional complexity)

    def generate_imports(self) -> str:
        """
        Generate the import statements for the generated resources file.

        Returns:
            str: The import statements.
        """
        # List of import statements
        imports = [
            "import botocore",
            "import datetime",
            "import time",
            "import functools",
            "from pydantic import validate_call, ConfigDict, BaseModel",
            "from typing import Dict, List, Literal, Optional, Union, Any\n"
            "from boto3.session import Session",
            "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, ResourceIterator, Unassigned, get_textual_rich_logger, "
            "snake_to_pascal, pascal_to_snake, is_not_primitive, is_not_str_dict, 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.logs import MultiLogStreamHandler",
            "from sagemaker_core.main.exceptions import *",
            "import sagemaker_core.main.shapes as shapes",
        ]

        formated_imports = "\n".join(imports)
        formated_imports += "\n\n"

        # Join the import statements with a newline character and return
        return formated_imports