Part of the KafkaBatcher.Collector implementation not related to GenServer behavior.
@spec store_partition_count(KafkaBatcher.Collector.State.t()) :: KafkaBatcher.Collector.State.t()