apply swiftlint corrections from #117

Author: jessesquires

Sources/CellViewModel.swift

@@ -74,9 +74,8 @@ extension CellViewModel {
     public func eraseToAnyViewModel() -> AnyCellViewModel {
         if let erasedViewModel = self as? AnyCellViewModel {
             return erasedViewModel
-        } else {
-            return AnyCellViewModel(self)
         }
+        return AnyCellViewModel(self)
     }
 
     // MARK: Internal
@@ -147,7 +146,7 @@ public struct AnyCellViewModel: CellViewModel {
     ///
     /// - Parameter viewModel: The view model to type-erase.
     public init<T: CellViewModel>(_ viewModel: T) {
-        if let erasedViewModel = viewModel as? AnyCellViewModel {
+        if let erasedViewModel = viewModel as? Self {
             self = erasedViewModel
             return
         }

Sources/SupplementaryViewModel.swift

@@ -40,9 +40,8 @@ extension SupplementaryViewModel {
     public func eraseToAnyViewModel() -> AnySupplementaryViewModel {
         if let erasedViewModel = self as? AnySupplementaryViewModel {
             return erasedViewModel
-        } else {
-            return AnySupplementaryViewModel(self)
         }
+        return AnySupplementaryViewModel(self)
     }
 
     // MARK: Internal
@@ -111,7 +110,7 @@ public struct AnySupplementaryViewModel: SupplementaryViewModel {
     ///
     /// - Parameter viewModel: The view model to type-erase.
     public init<T: SupplementaryViewModel>(_ viewModel: T) {
-        if let erasedViewModel = viewModel as? AnySupplementaryViewModel {
+        if let erasedViewModel = viewModel as? Self {
             self = erasedViewModel
             return
         }