code: def set_mocked_modules(cls, pe: PackageExporter) -> None:
      pe.mock(
          [
              "markupsafe._speedups",
          ]
      )
  
file_path: torchrec/examples/inference/dlrm_packager.py
imports: import argparse
  import sys
  from typing import List
  from dlrm_predict import DLRMModelConfig, DLRMPredictFactory
  from torch.package import PackageExporter
  from torchrec.datasets.criteo import DEFAULT_CAT_NAMES, DEFAULT_INT_NAMES
  from torchrec.inference.model_packager import PredictFactoryPackager
label_desc: Use image node_id 2 for calling the DLRMPredictFactoryPackager obj's underlying member method code with example usage: obj.set_mocked_modules(cls, pe) without return types