code: def context():
      frozen._ENSURE_FROZEN_STRICT = False
      yield
      frozen._ENSURE_FROZEN_STRICT = True
  
imports: import random
  import pytest
  import torch
  import torch.nn
  from torch.utils.data import TensorDataset, DataLoader
  import nni
  from nni.mutable import frozen
  from nni.nas.hub.pytorch.nasbench201 import OPS_WITH_STRIDE
  from nni.nas.nn.pytorch import Repeat
  from nni.nas.oneshot.pytorch.supermodule.base import BaseSuperNetModule
  from nni.nas.oneshot.pytorch.supermodule.proxyless import ProxylessMixedLayer, ProxylessMixedInput, ProxylessMixedRepeat, _iter_tensors
  from pytorch_lightning import LightningModule, Trainer
label_desc: Use image node_id 1 for calling a global function with example usage: context() without return types