update LocalEvaluationClient.kt to use FLAG_TYPE constants

amplitude · Aug 28, 2023 · d9af032 · d9af032
1 parent befd64c
commit d9af032
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/main/kotlin/LocalEvaluationClient.kt b/src/main/kotlin/LocalEvaluationClient.kt
@@ -5,9 +5,7 @@ import com.amplitude.experiment.assignment.AmplitudeAssignmentService
 import com.amplitude.experiment.assignment.Assignment
 import com.amplitude.experiment.assignment.AssignmentService
 import com.amplitude.experiment.assignment.InMemoryAssignmentFilter
-import com.amplitude.experiment.evaluation.EvaluationEngine
-import com.amplitude.experiment.evaluation.EvaluationEngineImpl
-import com.amplitude.experiment.evaluation.FlagResult
+import com.amplitude.experiment.evaluation.*
 import com.amplitude.experiment.evaluation.serialization.SerialVariant
 import com.amplitude.experiment.flag.FlagConfigApiImpl
 import com.amplitude.experiment.flag.FlagConfigService
@@ -62,7 +60,7 @@ class LocalEvaluationClient internal constructor(
             val isVariant = !entry.value.isDefaultVariant
             val isIncluded = (flagKeys.isEmpty() || flagKeys.contains(entry.key))
             val isDeployed = entry.value.deployed
-            if (isIncluded || entry.value.type == "mutual-exclusion-group" || entry.value.type == "holdout-group") {
+            if (isIncluded || entry.value.type == FLAG_TYPE_MUTUAL_EXCLUSION_GROUP || entry.value.type == FLAG_TYPE_HOLDOUT_GROUP) {
                 assignmentResults[entry.key] = entry.value
             }
             isVariant && isIncluded && isDeployed