Merge pull request #22 from Drusy/master

TRMosaicLayout delegate is now weak
vinnyoodles · Apr 13, 2018 · 79df860 · 79df860
2 parents a85faa7 + 8a8a32c
commit 79df860
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/TRMosaicLayout/Classes/TRMosaicLayout.swift b/TRMosaicLayout/Classes/TRMosaicLayout.swift
@@ -13,7 +13,7 @@ public enum TRMosaicCellType {
     case small
 }
 
-public protocol TRMosaicLayoutDelegate {
+public protocol TRMosaicLayoutDelegate: class {
 
     func collectionView(_ collectionView:UICollectionView, mosaicCellSizeTypeAtIndexPath indexPath:IndexPath) -> TRMosaicCellType
 
@@ -24,7 +24,7 @@ public protocol TRMosaicLayoutDelegate {
 
 open class TRMosaicLayout: UICollectionViewLayout {
 
-    open var delegate:TRMosaicLayoutDelegate!
+    open weak var delegate:TRMosaicLayoutDelegate!
 
     var columns = TRMosaicColumns()