code: def get_module_order(root_module):
      key2order = {}
  
      def apply(m):
          for name, child in m.named_children():
              if isinstance(child, Mutable):
                  key2order[child.key] = child.order
              else:
                  apply(child)
  
      apply(root_module)
      return key2order
  
imports: import logging
  from collections import OrderedDict
  import time
  import numpy
  import torch
  import nni.retiarii.nn.pytorch
  from nni.nas.pytorch.mutables import Mutable, InputChoice, LayerChoice
label_desc: Use image node_id 7 for calling a global function with example usage: get_module_order(root_module) and returns: key2order