def bbox_wrapper(func):
  """Adds a bboxes function argument to func and returns unchanged bboxes."""
  def wrapper(images, bboxes, *args, **kwargs):
    return (func(images, *args, **kwargs), bboxes)
  return wrapper
Adds a bboxes function argument to func and returns unchanged bboxes.
inference_api/src/main/object_detection/utils/autoaugment_utils.py