code: def test_complexity(key):
      """Measure the complexity of a single model."""
      reset_cfg()
      cfg_file = os.path.join(_PYCLS_DIR, key)
      load_cfg(cfg_file)
      return net.complexity(builders.get_model())
  
imports: import datetime
  import json
  import os
  import shutil
  import tempfile
  import unittest
  import pycls.core.builders
  import pycls.core.distributed
  import pycls.core.logging
  import pycls.core.net
  import pycls.core.trainer
  import pycls.models.model_zoo
  from parameterized import parameterized
  from pycls.core.config import cfg, load_cfg, reset_cfg
label_desc: Use image node_id 1 for calling a global function with example usage: test_complexity(key) and returns: net