fix 'type' attribute in evaluation_result accessed incorrectly

amplitude · Aug 5, 2023 · 3d02322 · 3d02322
1 parent 77b8ac5
commit 3d02322
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/amplitude_experiment/local/client.py b/src/amplitude_experiment/local/client.py
@@ -82,8 +82,8 @@ def evaluate(self, user: User, flag_keys: List[str] = None) -> Dict[str, Variant
             included = not filter_result or key in flag_keys
             if not value.get('isDefaultVariant') and included:
                 variants[key] = Variant(value['variant'].get('key'), value['variant'].get('payload'))
-            if included or evaluation_result[key].type == FLAG_TYPE_MUTUAL_EXCLUSION_GROUP or \
-                    evaluation_result[key].type == FLAG_TYPE_HOLDOUT_GROUP:
+            if included or evaluation_result[key]['type'] == FLAG_TYPE_MUTUAL_EXCLUSION_GROUP or \
+                    evaluation_result[key]['type'] == FLAG_TYPE_HOLDOUT_GROUP:
                 assignment_result[key] = evaluation_result[key]
 
         if self.assignment_service: