code: def get_drop(*keys):
      """Drops the given keys."""
  
      def _drop(data):
          return {k: v for k, v in data.items() if k not in keys}
  
      return _drop
  
imports: from typing import Optional, Tuple
  import numpy
  from scenic.dataset_lib.big_transfer.preprocessing import autoaugment
  from scenic.dataset_lib.big_transfer.preprocessing import utils
  from scenic.dataset_lib.big_transfer.registry import Registry
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_addons import image
label_desc: Use image node_id 32 for calling a global function with example usage: get_drop() and returns: _drop, unknown