From d9af03258ba263ccf9f6dee5d51f029bd43ca484 Mon Sep 17 00:00:00 2001 From: Tim Yiu Date: Mon, 28 Aug 2023 14:00:03 -0700 Subject: [PATCH] update LocalEvaluationClient.kt to use FLAG_TYPE constants --- src/main/kotlin/LocalEvaluationClient.kt | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/src/main/kotlin/LocalEvaluationClient.kt b/src/main/kotlin/LocalEvaluationClient.kt index 9ee6938..c86d7a3 100644 --- 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