package com.cinecor.android.common.viewmodel

import android.arch.lifecycle.ViewModel
import android.arch.lifecycle.ViewModelProvider
import com.cinecor.android.data.source.CinemasRepository
import javax.inject.Inject

@Suppress("UNCHECKED_CAST")
class CinemaViewModelFactory
@Inject constructor(private val repository: CinemasRepository) : ViewModelProvider.Factory {

    override fun <T : ViewModel> create(modelClass: Class<T>): T = CinemaViewModel(repository) as T
}
