docstr: Return all supernet modules in the model space.
code: def supernet_modules(self) -> Iterable[BaseSuperNetModule]:
      """Return all supernet modules in the model space."""
      for module in self.modules():
          if isinstance(module, BaseSuperNetModule):
              yield module
  
imports: from __future__ import annotations
  import warnings
  from typing import Any, Iterable, List, cast, TYPE_CHECKING
  import torch.optim
  import torch.nn
  from torch.optim import Optimizer
  import nni.nas.nn.pytorch
  from nni.nas.evaluator.pytorch import LightningModule, Trainer
  from nni.mutable import Sample
  from .supermodule.base import BaseSuperNetModule
label_desc: Use image node_id 3 for calling the BaseOneShotLightningModule obj's underlying member method code with example usage: obj.supernet_modules() without return types