code: def with_config(**overrides):
      """Decorator used to run a wrapped function with overriden config."""
  
      def decorator(f):
          def wrapper(*args, **kwargs):
              with assign(**overrides):
                  return f(*args, **kwargs)
  
          return wrapper
  
      return decorator
  
file_path: dm-haiku/haiku/_src/config.py
imports: import contextlib
  import dataclasses
  import threading
  from typing import Optional
label_desc: Use image node_id 5 for calling a global function with example usage: with_config() and returns: decorator, wrapper, f