From 1fa8a1da20026cf93611cff4d4fe8387d04c53cf Mon Sep 17 00:00:00 2001
From: Bryan Keller <kellerbryan19@gmail.com>
Date: Fri, 8 Mar 2024 11:37:48 -0800
Subject: [PATCH] Clear cached layout attributes if the data source counts
 change

---
 MagazineLayout/Public/MagazineLayout.swift | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/MagazineLayout/Public/MagazineLayout.swift b/MagazineLayout/Public/MagazineLayout.swift
index 343655c..029454c 100755
--- a/MagazineLayout/Public/MagazineLayout.swift
+++ b/MagazineLayout/Public/MagazineLayout.swift
@@ -808,6 +808,13 @@ public final class MagazineLayout: UICollectionViewLayout {
     hasDataSourceCountInvalidationBeforeReceivingUpdateItems = context.invalidateDataSourceCounts &&
       !context.invalidateEverything
 
+    if context.invalidateDataSourceCounts {
+      itemLayoutAttributes.removeAll()
+      headerLayoutAttributes.removeAll()
+      footerLayoutAttributes.removeAll()
+      backgroundLayoutAttributes.removeAll()
+    }
+
     super.invalidateLayout(with: context)
   }