diff --git a/src/main.py b/src/main.py index a27afff..9ddd7c0 100644 --- a/src/main.py +++ b/src/main.py @@ -35,7 +35,6 @@ if __name__ == '__main__': # (After step P1) Select 500 most popular users and 500 random users who meet a particular # criteria as our sample, also find news channels # select_user_sample() - get_english_news_channels() # Just curious, who are the 20 most popular individuals on twitter? # print(tabulate(((u.username, u.popularity) for u in load_user_sample().most_popular[:20]), diff --git a/src/process/twitter_process.py b/src/process/twitter_process.py index dea068b..2f1133c 100644 --- a/src/process/twitter_process.py +++ b/src/process/twitter_process.py @@ -177,7 +177,8 @@ def load_user_sample() -> Sample: """ j = json.loads(read(f'{USER_DIR}/processed/sample.json')) return Sample([ProcessedUser(*u) for u in j['most_popular']], - [ProcessedUser(*u) for u in j['random']]) + [ProcessedUser(*u) for u in j['random']], + j['english_news']) class Posting(NamedTuple):