DK
func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, referenceSizeForHeaderInSection section: Int)
Не вызывая
collectionView.reloadData()
? Нужно высоту header'a обновить после определенного экшена