Manually add population for national data (#1938)

Author: rzats

nchs_mortality/delphi_nchs_mortality/pull.py

@@ -127,6 +127,9 @@ def pull_nchs_mortality_data(token: str, test_file: Optional[str]=None):
                                     geocode_col="state", dropna=False)
     df = gmpr.add_geocode(df, "state_name", "state_id",
                           from_col="state", new_col="geo_id", dropna=False)
-    # Manually set geo_id for national data
+    # Manually set geo_id and population for national data
+    national_pop = gmpr.get_crosswalk("nation", "pop")
+    us_pop = national_pop.loc[national_pop["nation"] == "us"]["pop"][0]
+    df.loc[df["state"] == "United States", "population"] = us_pop
     df.loc[df["state"] == "United States", "geo_id"] = "us"
     return df[keep_columns]