code: def register_model_wrapper(name: str) -> Callable[[Type], Type]:
      """
      Register a model wrapper so that it is available via the CLI.
  
      >>> @register_model_wrapper("my_model_name")
      ... class MyModelWrapper:
      ...     pass
      """
  
      def _inner(cls_):
          global MODEL_WRAPPER_REGISTRY
          MODEL_WRAPPER_REGISTRY[name] = cls_
          return cls_
  
      return _inner
  
imports: import projects.safety_bench.model_wrappers
  import importlib
  import pkgutil
  from typing import Callable, Dict, Type
label_desc: Use image node_id 1 for calling a global function with example usage: register_model_wrapper(name) and returns: _inner, cls_