diff --git a/birdxplorer/models.py b/birdxplorer/models.py index 9ab46a3..f35f111 100644 --- 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