Merge pull request #2 from ChaosTong/patch-1

fix crash when data array count down
NicolasKim · Sep 13, 2017 · 7a54f3f · 7a54f3f
2 parents 2286a9f + 3f73236
commit 7a54f3f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ELWaterFallLayout/Classes/ELWaterFlowLayout.swift b/ELWaterFallLayout/Classes/ELWaterFlowLayout.swift
@@ -96,6 +96,7 @@ open class ELWaterFlowLayout: UICollectionViewFlowLayout {
         super.prepare()
         resetLineWidth()
         resetLineHeight()
+        attrArr.removeAll()
 
         if let sectionCount = self.collectionView?.dataSource?.numberOfSections!(in: self.collectionView!) {
             for section in 0..<sectionCount{