Merge pull request #48 from jenno-verdonck/patch-1

Fixes get_definitions failing when get_data wasn't called previously
socialfoundations · May 15, 2024 · ebace19 · ebace19
2 parents d78adc5 + 086003d
commit ebace19
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/folktables/load_acs.py b/folktables/load_acs.py
@@ -149,6 +149,8 @@ def load_definitions(root_dir, year=2018, horizon='1-Year', download=False):
     year_string = year if horizon == '1-Year' else f'{year - 4}-{year}'
     url = f'https://www2.census.gov/programs-surveys/acs/tech_docs/pums/data_dict/PUMS_Data_Dictionary_{year_string}.csv'
 
+    os.makedirs(base_datadir, exist_ok=True)
+
     response = requests.get(url)
     with open(file_path, 'wb') as handle:
         handle.write(response.content)