Merge pull request #26 from codeforjapan/feature/16-modeling-population

feat(models): update modelingPopulation type for UserEnrollment
codeforjapan · Nov 1, 2023 · 060fe7b · 060fe7b
2 parents 1cdbd28 + 9651750
commit 060fe7b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/birdxplorer/models.py b/birdxplorer/models.py
@@ -326,6 +326,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"]]
 UserEnrollmentLastEarnOutTimestamp = Union[TwitterTimestamp, Literal["1"]]
 
@@ -336,7 +341,7 @@ class UserEnrollment(BaseModel):
     successful_rating_needed_to_earn_in: NonNegativeInt
     timestamp_of_last_state_change: UserEnrollmentLastStateChangeTimeStamp
     timestamp_of_last_earn_out: UserEnrollmentLastEarnOutTimestamp
-    modeling_population: str
+    modeling_population: ModelingPopulation
     modeling_group: str