def __init__(
        self,
        model: torch.nn.Module,
    ) -> None:
        """
        Initializes the TorchModuleInspector.

        Parameters
        ----------
        model : torch.nn.Module
            The PyTorch model to be inspected.
        """
        if not isinstance(model, torch.nn.Module):
            raise TypeError('The "module" parameter must be of type "torch.nn.Module"')
        self._model = model

        Initializes the TorchModuleInspector.

        Parameters
        ----------
        model : torch.nn.Module
            The PyTorch model to be inspected.
        
coremltools/models/ml_program/experimental/torch/debugging_utils.py