code: def create_namespace(doctest_namespace):
      """
      Initialize namespace for doctest.
      Everything added to `doctest_namespace` will be available in the doctest.
      """
      from typing import Any, Dict, List, Set, Tuple, Union  # noqa
  
      import numpy as np  # noqa
      import opacus  # noqa
      import torch  # noqa
      from torch import nn  # noqa
  
      # Adding all imports in the doctest namespace
      doctest_namespace.update(**locals())
  
      doctest_namespace["MyCustomModel"] = MyCustomModel
      doctest_namespace["TensorDataset"] = TensorDataset
      doctest_namespace["demo_dataloader"] = create_demo_dataloader()
      doctest_namespace[
          "_init_private_training"
      ] = _init_private_training
  
file_path: opacus/conftest.py
imports: import pytest
  import torch
  from opacus import PrivacyEngine
  from torch import nn
  from torch.utils.data import DataLoader, TensorDataset
label_desc: Use image node_id 3 for calling a global function with example usage: create_namespace(doctest_namespace) without return types