Update AlignedCollectionViewFlowLayout.swift

Fix by mischa-hildebrand#35 (comment)
VladMelnik · Dec 30, 2021 · 369d5fe · 369d5fe
1 parent 49330ef
commit 369d5fe
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/AlignedCollectionViewFlowLayout/Classes/AlignedCollectionViewFlowLayout.swift b/AlignedCollectionViewFlowLayout/Classes/AlignedCollectionViewFlowLayout.swift
@@ -286,7 +286,12 @@ open class AlignedCollectionViewFlowLayout: UICollectionViewFlowLayout {
         var lineFrame = layoutAttributes.frame
         lineFrame.origin.x = sectionInset.left
         lineFrame.size.width = lineWidth
-        return super.layoutAttributesForElements(in: lineFrame) ?? []
+        let result = super.layoutAttributesForElements(in: lineFrame)
+        if let result = result, !result.isEmpty {
+            return result
+        } else {
+            return [UICollectionViewLayoutAttributes()]
+        }
     }
 
     /// Copmutes the alignment axis with which to align the items represented by the `layoutAttributes` objects vertically.