Register ViewModel Factory

RegisterViewModelFactory.kt

class RegisterViewModelFactory @Inject constructor(
    private val repository: UserRepository
): ViewModelProvider.Factory {

    @Suppress("UNCHECKED_CAST")
    override fun <T : ViewModel> create(modelClass: Class<T>): T {
        return if (modelClass.isAssignableFrom(RegisterViewModel::class.java!!)) {
            RegisterViewModel(repository) as T
        } else {
            throw IllegalArgumentException("ViewModel Not Found")
        }
    }
}