feat(models): update modelingPopulation type for UserEnrollment

codeforjapan · Oct 31, 2023 · 6fc79a6 · 6fc79a6
1 parent bd441a2
commit 6fc79a6
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/birdxplorer/models.py b/birdxplorer/models.py
@@ -286,6 +286,11 @@ class EnrollmentState(str, Enum):
     earned_out_no_acknowledge = "earnedOutNoAcknowledge"
 
 
+class ModelingPopulation(str, Enum):
+    control = "CORE"
+    treatment = "EXPANSION"
+
+
 UserEnrollmentLastStateChangeTimeStamp = Union[TwitterTimestamp, Literal["0"], Literal["103308100"]]
 
 
@@ -295,7 +300,7 @@ class UserEnrollment(BaseModel):
     successful_rating_needed_to_earn_in: str
     timestamp_of_last_state_change: UserEnrollmentLastStateChangeTimeStamp
     timestamp_of_last_earn_out: str
-    modeling_population: str
+    modeling_population: ModelingPopulation
     modeling_group: str