A function that takes an original data set and keeps all
data for all groups that are not-yet-treated by period tp as well
as for group g.
In particular, this keeps more data than functions like two_by_two
subset that use a fixed base period.
A main use case for this function is the interactive fixed effects approach proposed in Callaway and Tsyawo (2023).