docs: def mark(self, partition, offset):
          """
          Set the high-water mark in the current context.
  
          In order to know the current partition, it is helpful to initialize
          the consumer to provide partition info via:
  
          .. code:: python
  
              consumer.provide_partition_info()
  
          """
          max_offset = max(offset + 1, self.high_water_mark.get(partition, 0))
  
          self.logger.debug("Setting high-water mark to: %s",
                            {partition: max_offset})
  
          self.high_water_mark[partition] = max_offset