Add view model factory class

app/src/main/AndroidManifest.xml

@@ -40,8 +40,7 @@
 
         <activity
             android:name=".ui.login.LoginActivity"
-            android:launchMode="singleTop"
-            android:screenOrientation="portrait" />
+            android:launchMode="singleTop" />
 
         <activity
             android:name=".ui.main.MainActivity"
@@ -50,8 +49,7 @@
         <activity
             android:name="com.mindorks.framework.mvvm.ui.feed.FeedActivity"
             android:launchMode="singleTop"
-            android:parentActivityName=".ui.main.MainActivity"
-            android:screenOrientation="portrait">
+            android:parentActivityName=".ui.main.MainActivity">
 
             <!-- Parent activity meta-data to support 4.0 and lower -->
             <meta-data

app/src/main/java/com/mindorks/framework/mvvm/ViewModelProviderFactory.java

@@ -2,26 +2,64 @@ package com.mindorks.framework.mvvm;
 
 import android.arch.lifecycle.ViewModel;
 import android.arch.lifecycle.ViewModelProvider;
+import com.mindorks.framework.mvvm.data.DataManager;
+import com.mindorks.framework.mvvm.ui.about.AboutViewModel;
+import com.mindorks.framework.mvvm.ui.feed.FeedViewModel;
+import com.mindorks.framework.mvvm.ui.feed.blogs.BlogViewModel;
+import com.mindorks.framework.mvvm.ui.feed.opensource.OpenSourceViewModel;
+import com.mindorks.framework.mvvm.ui.login.LoginViewModel;
+import com.mindorks.framework.mvvm.ui.main.MainViewModel;
+import com.mindorks.framework.mvvm.ui.main.rating.RateUsViewModel;
+import com.mindorks.framework.mvvm.ui.splash.SplashViewModel;
+import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
 /**
- * Created by RV on 19/07/17.
- * <p>
- * A provider factory that persists ViewModels {@link ViewModel}.
- * Used if the view model has a parameterized constructor.
+ * Created by jyotidubey on 22/02/19.
  */
-public class ViewModelProviderFactory<V> implements ViewModelProvider.Factory {
 
-    private V viewModel;
+public class ViewModelProviderFactory extends ViewModelProvider.NewInstanceFactory {
 
-    public ViewModelProviderFactory(V viewModel) {
-        this.viewModel = viewModel;
-    }
+  private final DataManager dataManager;
+  private final SchedulerProvider schedulerProvider;
+
+  public ViewModelProviderFactory(DataManager dataManager,
+      SchedulerProvider schedulerProvider) {
+    this.dataManager = dataManager;
+    this.schedulerProvider = schedulerProvider;
+  }
 
-    @Override
-    public <T extends ViewModel> T create(Class<T> modelClass) {
-        if (modelClass.isAssignableFrom(viewModel.getClass())) {
-            return (T) viewModel;
-        }
-        throw new IllegalArgumentException("Unknown class name");
+
+  @Override
+  public <T extends ViewModel> T create(Class<T> modelClass) {
+    if (modelClass.isAssignableFrom(AboutViewModel.class)) {
+      //noinspection unchecked
+      return (T) new AboutViewModel(dataManager,schedulerProvider);
+    } else if (modelClass.isAssignableFrom(FeedViewModel.class)) {
+      //noinspection unchecked
+      return (T) new FeedViewModel(dataManager,schedulerProvider);
+    } else if (modelClass.isAssignableFrom(LoginViewModel.class)) {
+      //noinspection unchecked
+      return (T) new LoginViewModel(dataManager,schedulerProvider);
+    } else if (modelClass.isAssignableFrom(MainViewModel.class)) {
+      //noinspection unchecked
+      return (T) new MainViewModel(dataManager,schedulerProvider);
+    }
+    else if (modelClass.isAssignableFrom(BlogViewModel.class)) {
+      //noinspection unchecked
+      return (T) new BlogViewModel(dataManager,schedulerProvider);
+    }
+    else if (modelClass.isAssignableFrom(RateUsViewModel.class)) {
+      //noinspection unchecked
+      return (T) new RateUsViewModel(dataManager,schedulerProvider);
+    }
+    else if (modelClass.isAssignableFrom(OpenSourceViewModel.class)) {
+      //noinspection unchecked
+      return (T) new OpenSourceViewModel(dataManager,schedulerProvider);
+    }
+    else if (modelClass.isAssignableFrom(SplashViewModel.class)) {
+      //noinspection unchecked
+      return (T) new SplashViewModel(dataManager,schedulerProvider);
     }
-}
+    throw new IllegalArgumentException("Unknown ViewModel class: " + modelClass.getName());
+  }
+}
\ No newline at end of file

app/src/main/java/com/mindorks/framework/mvvm/ui/about/AboutFragment.java

@@ -16,10 +16,12 @@
 
 package com.mindorks.framework.mvvm.ui.about;
 
+import android.arch.lifecycle.ViewModelProviders;
 import android.os.Bundle;
 import android.support.annotation.Nullable;
 
 import com.mindorks.framework.mvvm.BR;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.FragmentAboutBinding;
 import com.mindorks.framework.mvvm.ui.base.BaseFragment;
@@ -34,6 +36,7 @@ public class AboutFragment extends BaseFragment<FragmentAboutBinding, AboutViewM
 
     public static final String TAG = AboutFragment.class.getSimpleName();
     @Inject
+    ViewModelProviderFactory factory;
     AboutViewModel mAboutViewModel;
 
     public static AboutFragment newInstance() {
@@ -55,6 +58,7 @@ public class AboutFragment extends BaseFragment<FragmentAboutBinding, AboutViewM
 
     @Override
     public AboutViewModel getViewModel() {
+        mAboutViewModel = ViewModelProviders.of(this,factory).get(AboutViewModel.class);
         return mAboutViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/about/AboutFragmentModule.java

@@ -16,6 +16,7 @@
 
 package com.mindorks.framework.mvvm.ui.about;
 
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.data.DataManager;
 import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
@@ -27,9 +28,8 @@ import dagger.Provides;
  */
 @Module
 public class AboutFragmentModule {
-
     @Provides
-    AboutViewModel provideAboutViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
-        return new AboutViewModel(dataManager, schedulerProvider);
+    ViewModelProviderFactory provideSplashViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 }

app/src/main/java/com/mindorks/framework/mvvm/ui/feed/FeedActivity.java

@@ -16,6 +16,7 @@
 
 package com.mindorks.framework.mvvm.ui.feed;
 
+import android.arch.lifecycle.ViewModelProviders;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Bundle;
@@ -26,6 +27,7 @@ import android.support.v4.app.TaskStackBuilder;
 import android.view.MenuItem;
 
 import com.mindorks.framework.mvvm.BR;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.ActivityFeedBinding;
 import com.mindorks.framework.mvvm.ui.base.BaseActivity;
@@ -35,6 +37,7 @@ import javax.inject.Inject;
 import dagger.android.AndroidInjector;
 import dagger.android.DispatchingAndroidInjector;
 import dagger.android.support.HasSupportFragmentInjector;
+import javax.inject.Named;
 
 /**
  * Created by amitshekhar on 10/07/17.
@@ -45,10 +48,12 @@ public class FeedActivity extends BaseActivity<ActivityFeedBinding, FeedViewMode
     @Inject
     DispatchingAndroidInjector<Fragment> fragmentDispatchingAndroidInjector;
     @Inject
-    FeedViewModel mFeedViewModel;
-    @Inject
     FeedPagerAdapter mPagerAdapter;
+    @Inject
+    @Named("FeedActivity")
+    ViewModelProviderFactory factory;
     private ActivityFeedBinding mActivityFeedBinding;
+    FeedViewModel mFeedViewModel;
 
     public static Intent newIntent(Context context) {
         return new Intent(context, FeedActivity.class);
@@ -66,6 +71,7 @@ public class FeedActivity extends BaseActivity<ActivityFeedBinding, FeedViewMode
 
     @Override
     public FeedViewModel getViewModel() {
+        mFeedViewModel = ViewModelProviders.of(this,factory).get(FeedViewModel.class);
         return mFeedViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/feed/FeedActivityModule.java

@@ -16,11 +16,13 @@
 
 package com.mindorks.framework.mvvm.ui.feed;
 
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.data.DataManager;
 import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
 import dagger.Module;
 import dagger.Provides;
+import javax.inject.Named;
 
 /**
  * Created by amitshekhar on 14/09/17.
@@ -34,8 +36,8 @@ public class FeedActivityModule {
     }
 
     @Provides
-    FeedViewModel provideFeedViewModel(DataManager dataManager,
-                                       SchedulerProvider schedulerProvider) {
-        return new FeedViewModel(dataManager, schedulerProvider);
+    @Named("FeedActivity")
+    ViewModelProviderFactory provideFeedSourceFactory(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 }

app/src/main/java/com/mindorks/framework/mvvm/ui/feed/blogs/BlogFragment.java

@@ -16,7 +16,6 @@
 
 package com.mindorks.framework.mvvm.ui.feed.blogs;
 
-import android.arch.lifecycle.ViewModelProvider;
 import android.arch.lifecycle.ViewModelProviders;
 import android.os.Bundle;
 import android.support.annotation.NonNull;
@@ -26,6 +25,7 @@ import android.support.v7.widget.LinearLayoutManager;
 import android.view.View;
 
 import com.mindorks.framework.mvvm.BR;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.data.model.api.BlogResponse;
 import com.mindorks.framework.mvvm.databinding.FragmentBlogBinding;
@@ -48,7 +48,7 @@ public class BlogFragment extends BaseFragment<FragmentBlogBinding, BlogViewMode
     @Inject
     LinearLayoutManager mLayoutManager;
     @Inject
-    ViewModelProvider.Factory mViewModelFactory;
+    ViewModelProviderFactory factory;
     private BlogViewModel mBlogViewModel;
 
     public static BlogFragment newInstance() {
@@ -70,7 +70,7 @@ public class BlogFragment extends BaseFragment<FragmentBlogBinding, BlogViewMode
 
     @Override
     public BlogViewModel getViewModel() {
-        mBlogViewModel = ViewModelProviders.of(this, mViewModelFactory).get(BlogViewModel.class);
+        mBlogViewModel = ViewModelProviders.of(this, factory).get(BlogViewModel.class);
         return mBlogViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/feed/blogs/BlogFragmentModule.java

@@ -16,7 +16,6 @@
 
 package com.mindorks.framework.mvvm.ui.feed.blogs;
 
-import android.arch.lifecycle.ViewModelProvider;
 import android.support.v7.widget.LinearLayoutManager;
 
 import com.mindorks.framework.mvvm.ViewModelProviderFactory;
@@ -46,8 +45,8 @@ public class BlogFragmentModule {
     }
 
     @Provides
-    ViewModelProvider.Factory provideBlogViewModel(BlogViewModel blogViewModel) {
-        return new ViewModelProviderFactory<>(blogViewModel);
+    ViewModelProviderFactory provideBlogViewModelFactory(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 
     @Provides

app/src/main/java/com/mindorks/framework/mvvm/ui/feed/opensource/OpenSourceFragment.java

@@ -17,7 +17,6 @@
 package com.mindorks.framework.mvvm.ui.feed.opensource;
 
 
-import android.arch.lifecycle.ViewModelProvider;
 import android.arch.lifecycle.ViewModelProviders;
 import android.os.Bundle;
 import android.support.annotation.NonNull;
@@ -27,6 +26,7 @@ import android.support.v7.widget.LinearLayoutManager;
 import android.view.View;
 
 import com.mindorks.framework.mvvm.BR;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.FragmentOpenSourceBinding;
 import com.mindorks.framework.mvvm.ui.base.BaseFragment;
@@ -46,7 +46,7 @@ public class OpenSourceFragment extends BaseFragment<FragmentOpenSourceBinding,
     @Inject
     OpenSourceAdapter mOpenSourceAdapter;
     @Inject
-    ViewModelProvider.Factory mViewModelFactory;
+    ViewModelProviderFactory factory;
     private OpenSourceViewModel mOpenSourceViewModel;
 
     public static OpenSourceFragment newInstance() {
@@ -68,7 +68,7 @@ public class OpenSourceFragment extends BaseFragment<FragmentOpenSourceBinding,
 
     @Override
     public OpenSourceViewModel getViewModel() {
-        mOpenSourceViewModel = ViewModelProviders.of(this, mViewModelFactory).get(OpenSourceViewModel.class);
+        mOpenSourceViewModel = ViewModelProviders.of(this, factory).get(OpenSourceViewModel.class);
         return mOpenSourceViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/feed/opensource/OpenSourceFragmentModule.java

@@ -16,7 +16,6 @@
 
 package com.mindorks.framework.mvvm.ui.feed.opensource;
 
-import android.arch.lifecycle.ViewModelProvider;
 import android.support.v7.widget.LinearLayoutManager;
 
 import com.mindorks.framework.mvvm.ViewModelProviderFactory;
@@ -48,7 +47,7 @@ public class OpenSourceFragmentModule {
     }
 
     @Provides
-    ViewModelProvider.Factory provideOpenSourceViewModel(OpenSourceViewModel openSourceViewModel) {
-        return new ViewModelProviderFactory<>(openSourceViewModel);
+    ViewModelProviderFactory provideOpenSourceFactory(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 }

app/src/main/java/com/mindorks/framework/mvvm/ui/login/LoginActivity.java

@@ -16,12 +16,14 @@
 
 package com.mindorks.framework.mvvm.ui.login;
 
+import android.arch.lifecycle.ViewModelProviders;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Bundle;
 import android.widget.Toast;
 
 import com.mindorks.framework.mvvm.BR;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.ActivityLoginBinding;
 import com.mindorks.framework.mvvm.ui.base.BaseActivity;
@@ -36,6 +38,7 @@ import javax.inject.Inject;
 public class LoginActivity extends BaseActivity<ActivityLoginBinding, LoginViewModel> implements LoginNavigator {
 
     @Inject
+    ViewModelProviderFactory factory;
     LoginViewModel mLoginViewModel;
     private ActivityLoginBinding mActivityLoginBinding;
 
@@ -55,6 +58,7 @@ public class LoginActivity extends BaseActivity<ActivityLoginBinding, LoginViewM
 
     @Override
     public LoginViewModel getViewModel() {
+        mLoginViewModel = ViewModelProviders.of(this,factory).get(LoginViewModel.class);
         return mLoginViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/login/LoginActivityModule.java

@@ -16,6 +16,7 @@
 
 package com.mindorks.framework.mvvm.ui.login;
 
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.data.DataManager;
 import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
@@ -29,7 +30,7 @@ import dagger.Provides;
 public class LoginActivityModule {
 
     @Provides
-    LoginViewModel provideLoginViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
-        return new LoginViewModel(dataManager, schedulerProvider);
+    ViewModelProviderFactory provideLoginViewModelFactory(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 }

app/src/main/java/com/mindorks/framework/mvvm/ui/main/MainActivity.java

@@ -16,7 +16,6 @@
 
 package com.mindorks.framework.mvvm.ui.main;
 
-import android.arch.lifecycle.ViewModelProvider;
 import android.arch.lifecycle.ViewModelProviders;
 import android.content.Context;
 import android.content.Intent;
@@ -38,6 +37,7 @@ import android.view.View;
 
 import com.mindorks.framework.mvvm.BR;
 import com.mindorks.framework.mvvm.BuildConfig;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.ActivityMainBinding;
 import com.mindorks.framework.mvvm.databinding.NavHeaderMainBinding;
@@ -55,13 +55,15 @@ import javax.inject.Inject;
 import dagger.android.AndroidInjector;
 import dagger.android.DispatchingAndroidInjector;
 import dagger.android.support.HasSupportFragmentInjector;
+import javax.inject.Named;
 
 public class MainActivity extends BaseActivity<ActivityMainBinding, MainViewModel> implements MainNavigator, HasSupportFragmentInjector {
 
     @Inject
     DispatchingAndroidInjector<Fragment> fragmentDispatchingAndroidInjector;
     @Inject
-    ViewModelProvider.Factory mViewModelFactory;
+    @Named("MainActivity")
+    ViewModelProviderFactory factory;
     private ActivityMainBinding mActivityMainBinding;
     private SwipePlaceHolderView mCardsContainerView;
     private DrawerLayout mDrawer;
@@ -86,7 +88,7 @@ public class MainActivity extends BaseActivity<ActivityMainBinding, MainViewMode
 
     @Override
     public MainViewModel getViewModel() {
-        mMainViewModel = ViewModelProviders.of(this, mViewModelFactory).get(MainViewModel.class);
+        mMainViewModel = ViewModelProviders.of(this, factory).get(MainViewModel.class);
         return mMainViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/main/MainActivityModule.java

@@ -16,14 +16,13 @@
 
 package com.mindorks.framework.mvvm.ui.main;
 
-import android.arch.lifecycle.ViewModelProvider;
-
 import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.data.DataManager;
 import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
 import dagger.Module;
 import dagger.Provides;
+import javax.inject.Named;
 
 /**
  * Created by amitshekhar on 14/09/17.
@@ -32,12 +31,9 @@ import dagger.Provides;
 public class MainActivityModule {
 
     @Provides
-    ViewModelProvider.Factory mainViewModelProvider(MainViewModel mainViewModel) {
-        return new ViewModelProviderFactory<>(mainViewModel);
+    @Named("MainActivity")
+    ViewModelProviderFactory provideMainSourceFactory(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 
-    @Provides
-    MainViewModel provideMainViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
-        return new MainViewModel(dataManager, schedulerProvider);
-    }
 }

app/src/main/java/com/mindorks/framework/mvvm/ui/main/rating/RateUsDialog.java

@@ -16,6 +16,7 @@
 
 package com.mindorks.framework.mvvm.ui.main.rating;
 
+import android.arch.lifecycle.ViewModelProviders;
 import android.databinding.DataBindingUtil;
 import android.os.Bundle;
 import android.support.annotation.NonNull;
@@ -24,6 +25,7 @@ import android.view.LayoutInflater;
 import android.view.View;
 import android.view.ViewGroup;
 
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.DialogRateUsBinding;
 import com.mindorks.framework.mvvm.ui.base.BaseDialog;
@@ -40,6 +42,7 @@ public class RateUsDialog extends BaseDialog implements RateUsCallback {
 
     private static final String TAG = RateUsDialog.class.getSimpleName();
     @Inject
+    ViewModelProviderFactory factory;
     RateUsViewModel mRateUsViewModel;
 
     public static RateUsDialog newInstance() {
@@ -60,7 +63,7 @@ public class RateUsDialog extends BaseDialog implements RateUsCallback {
         View view = binding.getRoot();
 
         AndroidSupportInjection.inject(this);
-
+        mRateUsViewModel = ViewModelProviders.of(this,factory).get(RateUsViewModel.class);
         binding.setViewModel(mRateUsViewModel);
 
         mRateUsViewModel.setNavigator(this);

app/src/main/java/com/mindorks/framework/mvvm/ui/main/rating/RateUsDialogModule.java

@@ -16,6 +16,7 @@
 
 package com.mindorks.framework.mvvm.ui.main.rating;
 
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.data.DataManager;
 import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
@@ -29,7 +30,7 @@ import dagger.Provides;
 public class RateUsDialogModule {
 
     @Provides
-    RateUsViewModel provideRateUsViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
-        return new RateUsViewModel(dataManager, schedulerProvider);
+    ViewModelProviderFactory provideRateUsViewModelFactory(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 }

app/src/main/java/com/mindorks/framework/mvvm/ui/splash/SplashActivity.java

@@ -16,11 +16,13 @@
 
 package com.mindorks.framework.mvvm.ui.splash;
 
+import android.arch.lifecycle.ViewModelProviders;
 import android.content.Context;
 import android.content.Intent;
 import android.os.Bundle;
 
 import com.mindorks.framework.mvvm.BR;
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.R;
 import com.mindorks.framework.mvvm.databinding.ActivitySplashBinding;
 import com.mindorks.framework.mvvm.ui.base.BaseActivity;
@@ -36,6 +38,7 @@ import javax.inject.Inject;
 public class SplashActivity extends BaseActivity<ActivitySplashBinding, SplashViewModel> implements SplashNavigator {
 
     @Inject
+    ViewModelProviderFactory factory;
     SplashViewModel mSplashViewModel;
 
     public static Intent newIntent(Context context) {
@@ -54,6 +57,7 @@ public class SplashActivity extends BaseActivity<ActivitySplashBinding, SplashVi
 
     @Override
     public SplashViewModel getViewModel() {
+        mSplashViewModel = ViewModelProviders.of(this,factory).get(SplashViewModel.class);
         return mSplashViewModel;
     }
 

app/src/main/java/com/mindorks/framework/mvvm/ui/splash/SplashActivityModule.java

@@ -16,6 +16,7 @@
 
 package com.mindorks.framework.mvvm.ui.splash;
 
+import com.mindorks.framework.mvvm.ViewModelProviderFactory;
 import com.mindorks.framework.mvvm.data.DataManager;
 import com.mindorks.framework.mvvm.utils.rx.SchedulerProvider;
 
@@ -29,7 +30,7 @@ import dagger.Provides;
 public class SplashActivityModule {
 
     @Provides
-    SplashViewModel provideSplashViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
-        return new SplashViewModel(dataManager, schedulerProvider);
+    ViewModelProviderFactory provideSplashViewModel(DataManager dataManager, SchedulerProvider schedulerProvider) {
+        return new ViewModelProviderFactory(dataManager, schedulerProvider);
     }
 }