Doens't work correctly after changing cell size programmatically.

[git-hwa]

Thank you for this awesome library for collection view.

I have integrated your library in my project, and I have one problem.
After calling UICollectionViewDelegateFlowLayout method - sizeForItemAt, the transitioning cell sizes are doesn't changed.

My code is here and there are written on both of first and second view controller.

extension MyViewController: UICollectionViewDelegateFlowLayout {
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        let desiredWidth = (self.view.frame.size.width - 2) / 2
        let desiredHeight = desiredWidth

        return CGSize(width: desiredWidth, height: desiredHeight)
    }
}

Please let me know how can I handle this problem.
Thank you

[git-hwa]

@CezaryKopacz Any solution for this issue?
It seems your example project also doesn't work with my approach.

[git-hwa]

I have fixed by setting itemSize manually.

extension MyViewController: UICollectionViewDelegateFlowLayout {
    func collectionView(_ collectionView: UICollectionView, layout collectionViewLayout: UICollectionViewLayout, sizeForItemAt indexPath: IndexPath) -> CGSize {
        let desiredWidth = (self.view.frame.size.width - 2) / 2
        let desiredHeight = desiredWidth

        (self.collectionViewLayout as! UICollectionViewFlowLayout).itemSize = CGSize(width: desiredWidth, height: desiredHeight)

        return CGSize(width: desiredWidth, height: desiredHeight)
    }
}