def __init__(
        self,
        module: ScriptModule,
        annotator: Optional[TorchScriptModuleAnnotator] = None,
    ):
        """
        Initializes the _TorchModuleInliner.

        Parameters
        ----------
        module : ScriptModule
            The root torch module to be inlined.

        annotator : Optional[_TorchScriptModuleAnnotator]
            An optional annotator for annotating and tracking the nodes in the graph.
        """
        self.module = module
        self.annotator = annotator

        Initializes the _TorchModuleInliner.

        Parameters
        ----------
        module : ScriptModule
            The root torch module to be inlined.

        annotator : Optional[_TorchScriptModuleAnnotator]
            An optional annotator for annotating and tracking the nodes in the graph.
        
coremltools/models/ml_program/experimental/torch/debugging_utils.py