zipcode-features 0.1.0__tar.gz → 0.1.2__tar.gz

{zipcode_features-0.1.0 → zipcode_features-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: zipcode-features
-Version: 0.1.0
+Version: 0.1.2
 Summary: A Python package to process and extract features from zipcode data.
 Project-URL: Homepage, https://github.com/EricSchles/zipcode_features
 Author-email: Eric Schles <your.email@example.com>

{zipcode_features-0.1.0 → zipcode_features-0.1.2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "zipcode-features"
-version = "0.1.0"
+version = "0.1.2"
 description = "A Python package to process and extract features from zipcode data."
 readme = "README.md"
 requires-python = ">=3.10"

{zipcode_features-0.1.0 → zipcode_features-0.1.2}/setup.py

@@ -10,7 +10,7 @@ README = (HERE / "README.md").read_text()
 # This call to setup() does all the work
 setup(
     name="zipcode_features",
-    version="0.0.9",
+    version="0.1.2",
     description="A tool to get features based on census data from zipcodes",
     long_description=README,
     long_description_content_type="text/markdown",

{zipcode_features-0.1.0 → zipcode_features-0.1.2}/src/zipcode_features/__init__.py

@@ -1,4 +1,4 @@
-__version__ = '0.1.0'
+__version__ = '0.1.2'
 
 import zipcodes
 from zipcode3.search import SearchEngine
@@ -170,12 +170,10 @@ def _get_cbsa_code_to_cbsa_name() -> dict:
     """
     with resources.path("zipcode_features.data", "cbsa_codes.json") as json_path:
         code_to_name = json.load(open(json_path))
-    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.split().str.join(' ')
-    return df.to_dict()
+    for code in code_to_name:
+        value = code_to_name[code].replace(" -", "-")
+        code_to_name[code] = " ".join([elem for elem in x.split(" ") if elem != ''])
+    return code_to_name
 
 def _get_zip_to_fips_code() -> dict:
     with resources.path("zipcode_features.data", "ZIP_COUNTY_122025.csv") as csv_path:
@@ -195,7 +193,9 @@ def _get_bls_data() -> pd.DataFrame:
         "bls_2025_quarter_two.csv"
     ]
     cols_to_drop = [
+        "own_code",
         "industry_code",
+        "agglvl_code",
         "size_code",
         "disclosure_code",
         'lq_disclosure_code',