Fix observation feeds favorite and important filters

ngageoint · Apr 24, 2024 · 631acb6 · 631acb6
1 parent 4277a24
commit 631acb6
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mage/src/main/java/mil/nga/giat/mage/newsfeed/ObservationFeedViewModel.kt b/mage/src/main/java/mil/nga/giat/mage/newsfeed/ObservationFeedViewModel.kt
@@ -21,6 +21,8 @@ import mil.nga.giat.mage.data.datasource.observation.ObservationLocalDataSource
 import mil.nga.giat.mage.database.model.observation.State
 import mil.nga.giat.mage.data.datasource.event.EventLocalDataSource
 import mil.nga.giat.mage.data.datasource.user.UserLocalDataSource
+import mil.nga.giat.mage.database.model.observation.ObservationFavorite
+import mil.nga.giat.mage.database.model.observation.ObservationImportant
 import mil.nga.giat.mage.sdk.event.IObservationEventListener
 import java.sql.SQLException
 import java.util.*
@@ -31,8 +33,8 @@ class ObservationFeedViewModel @Inject constructor(
    val application: Application,
    val preferences: SharedPreferences,
    private val observationDao: Dao<Observation, Long>,
-   private val observationImportantDao: Dao<Observation, Long>,
-   private val observationFavoriteDao: Dao<Observation, Long>,
+   private val observationImportantDao: Dao<ObservationImportant, Long>,
+   private val observationFavoriteDao: Dao<ObservationFavorite, Long>,
    private val observationLocalDataSource: ObservationLocalDataSource,
    private val observationRepository: ObservationRepository,
    private val userLocalDataSource: UserLocalDataSource,