A kotlin extension function that eliminates the need of writing ViewModelFactory

ViewModelFactoryExt.kt

import androidx.fragment.app.Fragment
import androidx.lifecycle.ViewModel
import androidx.lifecycle.ViewModelProvider


/**
* An extension function that removes the need to write ViewModelFactory by creating a factory based on ViewModel arguments
 *
 * Example:
 * private val viewModel: DailyForecastViewModel by DailyForecastViewModel(requireContext()).createViewModel(this)
 */
inline fun <reified VM : ViewModel> ViewModel.createViewModel(fragment: Fragment): Lazy<VM> {
    return lazy {
        ViewModelProvider(fragment.viewModelStore,
            object : ViewModelProvider.Factory {
                override fun <T : ViewModel?> create(modelClass: Class<T>): T {
                    @Suppress("UNCHECKED_CAST")
                    return this@createViewModel as T
                }
            }
        ).get(VM::class.java)
    }
}