<filename>src/cdk_stacks/sagemaker_app_stage.py
import aws_cdk as cdk
from constructs import Construct
from .sagemaker_stack import SagemakerStack


class PipelineAppStage(cdk.Stage):
    def __init__(
        self, scope: Construct, construct_id: str, config: dict, **kwargs
    ) -> None:
        super().__init__(scope, construct_id, **kwargs)

        sagemakerStack = SagemakerStack(self, "CDKModelSagemakerStack", config=config)
