zipcode-features 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

zipcode_features/__init__.py

@@ -165,9 +165,10 @@ def _get_zip_to_cbsa_code() -> dict:
     mapping is of the form:
     {"zip code": "cbsa code"}
     """
-    with resources.path("zipcode_features.data", "CBSA_ZIP_122025.csv") as csv_path:
+    with resources.path("zipcode_features.data", "cbsa_to_zip.csv") as csv_path:
         df = pd.read_csv(csv_path, dtype={'ZIP': str, "CBSA": str})
     df = correct_code(df, "ZIP")
+    df = correct_code(df, "CBSA")
     mapping = df.set_index('ZIP')['CBSA'].to_dict()
     return partial(map_zip_to_cbsa, mapping)
 
@@ -187,7 +188,7 @@ def _get_cbsa_code_to_cbsa_name() -> dict:
     df = pd.DataFrame(columns=["code", "name"])
     df["name"] = code_to_name.values()
     df["code"] = code_to_name.keys()
-    df["name"] = df["name"].str.replace(" -", "-")
+    df["name"] = df["name"].str.replace(" -", "-").replace("- ", "-")
     df["name"] = df["name"].str.split().str.join(' ')
     code_to_name = df.set_index('code')['name'].to_dict()
     return partial(map_cbsa_code_to_name, code_to_name)