code: def wraps(f, docs: bool = True):
      assignments = ("__annotations__",)
  
      if docs:
          assignments += ("__doc__",)
  
      return functools.wraps(f, assigned=assignments, updated=())
  
imports: import enum
  import functools
  import hashlib
  import inspect
  import io
  import os
  import re
  import shutil
  import sys
  import typing
  import urllib
  from functools import total_ordering
  import jax
  import jax.numpy
  import jax.tree_util
  import numpy
  import treex
  from elegy import types
label_desc: Use image node_id 1 for calling a global function with example usage: wraps(f, docs) and returns: functools